We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 381001a commit 5ca8491Copy full SHA for 5ca8491
src/sasctl/tasks.py
@@ -49,8 +49,8 @@ def _sklearn_to_dict(model):
49
'RandomForestClassifier': 'Forest',
50
'DecisionTreeClassifier': 'Decision tree',
51
'DecisionTreeRegressor': 'Decision tree',
52
- 'classifier': 'Classification',
53
- 'regressor': 'Prediction'}
+ 'classifier': 'classification',
+ 'regressor': 'prediction'}
54
55
if hasattr(model, '_final_estimator'):
56
estimator = type(model._final_estimator)
0 commit comments