提交 8d7e13cc 编写于 作者: M Moez Ali

updated tuning grid of dt in classification.py and regression.py

上级 fb5057b5
......@@ -4358,7 +4358,7 @@ def tune_model(estimator = None,
param_grid = custom_grid
else:
param_grid = {"max_depth": np.random.randint(1, (len(X_train.columns)*.85),20),
"max_features": np.random.randint(3, len(X_train.columns),20),
"max_features": np.random.randint(1, len(X_train.columns),20),
"min_samples_leaf": [2,3,4,5,6],
"criterion": ["gini", "entropy"],
}
......
......@@ -4802,7 +4802,7 @@ def tune_model(estimator,
else:
param_grid = {"max_depth": np.random.randint(1, (len(X_train.columns)*.85),20),
"max_features": np.random.randint(3, len(X_train.columns),20),
"max_features": np.random.randint(1, len(X_train.columns),20),
"min_samples_leaf": [2,3,4,5,6],
"criterion": ["mse", "mae", "friedman_mse"],
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册