Skip to content

Commit 5ca8491

Browse files
authored
revert
1 parent 381001a commit 5ca8491

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/sasctl/tasks.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,8 +49,8 @@ def _sklearn_to_dict(model):
4949
'RandomForestClassifier': 'Forest',
5050
'DecisionTreeClassifier': 'Decision tree',
5151
'DecisionTreeRegressor': 'Decision tree',
52-
'classifier': 'Classification',
53-
'regressor': 'Prediction'}
52+
'classifier': 'classification',
53+
'regressor': 'prediction'}
5454

5555
if hasattr(model, '_final_estimator'):
5656
estimator = type(model._final_estimator)

0 commit comments

Comments
 (0)