Skip to content

Commit bd1d52b

Browse files
authored
Fix missing cuML Random Forest parameter (#74)
1 parent dda3bac commit bd1d52b

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

cuml_bench/df_clsf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ def fit(X, y):
7272
n_estimators=params.num_trees,
7373
max_depth=params.max_depth,
7474
max_features=params.max_features,
75-
min_rows_per_node=params.min_samples_split,
75+
min_samples_split=params.min_samples_split,
7676
max_leaves=params.max_leaf_nodes,
7777
min_impurity_decrease=params.min_impurity_decrease,
7878
bootstrap=params.bootstrap)

cuml_bench/df_regr.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def fit(X, y):
7070
n_estimators=params.num_trees,
7171
max_depth=params.max_depth,
7272
max_features=params.max_features,
73-
min_rows_per_node=params.min_samples_split,
73+
min_samples_split=params.min_samples_split,
7474
max_leaves=params.max_leaf_nodes,
7575
min_impurity_decrease=params.min_impurity_decrease,
7676
bootstrap=params.bootstrap)

0 commit comments

Comments
 (0)