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 269783c commit 40793c4Copy full SHA for 40793c4
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