Skip to content

Commit dcf3308

Browse files
committed
FIX MAE is not supported
1 parent 8c5d2d2 commit dcf3308

3 files changed

Lines changed: 2 additions & 2 deletions

File tree

python/cuml/cuml/ensemble/randomforest_common.pyx

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -133,8 +133,6 @@ class BaseRandomForestModel(UniversalBase):
133133
else:
134134
if criterion == "squared_error":
135135
split_criterion = "mse"
136-
elif criterion == "absolute_error":
137-
split_criterion = "mae"
138136
elif criterion == "poisson":
139137
split_criterion = "poisson"
140138
elif criterion == "gini":

python/cuml/cuml/ensemble/randomforestclassifier.pyx

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -263,6 +263,7 @@ class RandomForestClassifier(BaseRandomForestModel,
263263
_hyperparam_interop_translator = {
264264
"criterion": {
265265
"log_loss": "NotImplemented",
266+
"absolute_error": "NotImplemented",
266267
},
267268
"oob_score": {
268269
True: "NotImplemented",

python/cuml/cuml/ensemble/randomforestregressor.pyx

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -265,6 +265,7 @@ class RandomForestRegressor(BaseRandomForestModel,
265265
_hyperparam_interop_translator = {
266266
"criterion": {
267267
"friedman_mse": "NotImplemented",
268+
"absolute_error": "NotImplemented",
268269
},
269270
"oob_score": {
270271
True: "NotImplemented",

0 commit comments

Comments
 (0)