Skip to content

Commit 8c5d2d2

Browse files
committed
Correct default
1 parent 0df2f3c commit 8c5d2d2

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

python/cuml/cuml/ensemble/randomforest_common.pyx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@ class BaseRandomForestModel(UniversalBase):
7373
classes_ = CumlArrayDescriptor()
7474

7575
@device_interop_preparation
76-
def __init__(self, *, split_criterion, n_streams=4, n_estimators=100,
76+
def __init__(self, *, split_criterion=None, n_streams=4, n_estimators=100,
7777
max_depth=16, handle=None, max_features='sqrt', n_bins=128,
7878
bootstrap=True,
7979
verbose=False, min_samples_leaf=1, min_samples_split=2,

python/cuml/cuml/ensemble/randomforestclassifier.pyx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -276,7 +276,7 @@ class RandomForestClassifier(BaseRandomForestModel,
276276
}
277277

278278
@device_interop_preparation
279-
def __init__(self, *, split_criterion=0, handle=None, verbose=False,
279+
def __init__(self, *, split_criterion=None, handle=None, verbose=False,
280280
output_type=None,
281281
**kwargs):
282282

0 commit comments

Comments
 (0)