Open
Description
Currently we have a functional API with static methods like build_model.build_model_from_featureset
and predict.model_predictions
. Not only are these a bit of a mouthful, they're also a bit unintuitive because of how differently they are used than the corresponding sklearn
functionality. I suggest that we try to modify our API to be more like that of sklearn
, specifically:
- Add a
Model
class that wraps asklearn
model and provides ansklearn
-like API (.fit
,.predict
,.predict_proba
) for interacting withFeatureset
objects. - (optional) Also change
featurize
to behave more like ansklearn
preprocessor; maybe you'd initialize aFeatureset
and call.transform
on a dataset? This is less obvious but there's probably some improvement to be made here as well.
Metadata
Metadata
Assignees
Labels
No labels