基于树的模型的最佳超参数优化

时间:2018-09-07 17:08:30

标签: python-3.x machine-learning scikit-learn decision-tree grid-search

我正在尝试生成5种机器学习模型,并根据网格搜索类对它们进行调整,以便以最佳方式调整模型,以便能够将其用于预测每天都会出现的新数据。问题是执行此操作所需的时间太长。因此,我的问题是,绝对需要什么级别的参数调整,但是不会花费2个小时以上的时间?以下是我使用的调整和分类器的代码:

#Training and Test Sets
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size = .20, 
random_state = 10)

#Classifiers 
dtc = DecisionTreeClassifier()
randf = RandomForestClassifier()
bag = BaggingClassifier()
gradb = GradientBoostingClassifier()
knn = KNeighborsClassifier()
ada = AdaBoostClassifier()

#Hyperparamter Tuning for the Models being used

#Scoring Criteria 
scoring = {'precision': make_scorer(precision_score), 'accuracy': 
make_scorer(accuracy_score)}

#Grid Search for the Decision Tree
param_dtc = {'min_samples_split': np.arange(2, 10), 'min_samples_leaf': 
np.arange(.05, .2), 'max_leaf_nodes': np.arange(2, 30)}
cv_dtc = GridSearchCV(estimator = dtc, param_grid = param_dtc, cv = 3, 
scoring = scoring, refit='precision', n_jobs=-1)
#Grid Search for the Random Forest Model  
param_randf = {'n_estimators': np.arange(10, 20), 'min_samples_split': 
np.arange(2, 10), 'min_samples_leaf': np.arange(.15, .33), 'max_leaf_nodes': 
np.arange(2, 30), 'bootstrap': ['True', 'False']}
cv_randf = GridSearchCV(estimator = randf, param_grid = param_randf, cv = 3, 
scoring = scoring, refit='precision', n_jobs=-1)
#Grid Search for the Bagging Model 
param_bag = {'n_estimators': np.arange(10, 30), 'max_samples': np.arange(2, 
30), 'bootstrap': ['True', 'False'], 'bootstrap_features': ['True', 
'False']}
cv_bag = GridSearchCV(estimator = bag, param_grid = param_bag, cv = 3, 
scoring = scoring, refit='precision', n_jobs=-1)
#Grid Search for the Gradient Boosting Model 
param_gradb = {'loss': ['deviance', 'exponential'], 'learning_rate': 
np.arange(.05, .1), 'max_depth': np.arange(2, 10), 'min_samples_split': 
np.arange(2, 10), 'min_samples_leaf': np.arange(.15, .33), 'max_leaf_nodes': 
np.arange(2, 30)}
cv_gradb = GridSearchCV(estimator = gradb, param_grid = param_gradb, cv = 3, 
scoring = scoring, refit='precision', n_jobs=-1)
#Grid Search for the Adaptive Boosting Model
param_ada = {'n_estimators': np.arange(10, 30), 'learning_rate': 
np.arange(.05, .1)}
cv_ada = GridSearchCV(estimator = ada, param_grid = param_ada, cv = 3, 
scoring = scoring, refit='precision', n_jobs=-1)

train_dict = {'dtc': cv_dtc.fit(x_train, y_train), 'randf': 
cv_randf.fit(x_train, y_train), 'bag': cv_bag.fit(x_train, y_train), 
'gradb': cv_gradb.fit(x_train, y_train), 'ada': cv_ada.fit(x_train, 
y_train)}

1 个答案:

答案 0 :(得分:0)

  1. 您可能会考虑进行一些迭代的网格搜索。例如,不要将'n_estimators'设置为np.arange(10,30),而是将其设置为[10,15,20,25,30]。是最优参数15,继续[11,13,15,17,19]。您将找到一种自动执行此过程的方法。这样可以节省很多时间。

  2. 播放数据。您正在调整许多超参数。在决策树中,“ min_samples_split”,“ min_samples_leaf”和“ max_leaf_nodes”的影响相互交集。定义所有这些对象可能没有必要。