diff --git a/cuml_bench/df_clsf.py b/cuml_bench/df_clsf.py index 429cdf741..e3b133544 100755 --- 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 index 410a189c8..62040791b 100644 --- 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)