Fix missing cuML Random Forest parameter (IntelPython#74)

asprasad · Jun 10, 2021 · bd1d52b · bd1d52b
1 parent dda3bac
commit bd1d52b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cuml_bench/df_clsf.py b/cuml_bench/df_clsf.py
@@ -72,7 +72,7 @@ def fit(X, y):
                                  n_estimators=params.num_trees,
                                  max_depth=params.max_depth,
                                  max_features=params.max_features,
-                                 min_rows_per_node=params.min_samples_split,
+                                 min_samples_split=params.min_samples_split,
                                  max_leaves=params.max_leaf_nodes,
                                  min_impurity_decrease=params.min_impurity_decrease,
                                  bootstrap=params.bootstrap)

diff --git a/cuml_bench/df_regr.py b/cuml_bench/df_regr.py
@@ -70,7 +70,7 @@ def fit(X, y):
                                  n_estimators=params.num_trees,
                                  max_depth=params.max_depth,
                                  max_features=params.max_features,
-                                 min_rows_per_node=params.min_samples_split,
+                                 min_samples_split=params.min_samples_split,
                                  max_leaves=params.max_leaf_nodes,
                                  min_impurity_decrease=params.min_impurity_decrease,
                                  bootstrap=params.bootstrap)