Open kwinkunks opened 1 year ago
API like:
df.redflag.cool_method(X=None, y=None, **kwargs)
E.g.
features = ['GR', 'RHOB', 'PE'] # Columns in df.
df.redflag.cool_method(X=features, y='Lithology') # y can also be a list.
Then...
X
and y
are None
, treat the whole df
as X
X
is None
, use y
and treat the remaining columns as X
y
is None
, use X
and ignore y
Could be interesting to implement detectors etc as methods on DataFrames, eg
How-to: https://pandas.pydata.org/docs/development/extending.html
Could do the same for xarrays I guess, but DataFrames are key.