【发布时间】:2019-02-13 01:35:56
【问题描述】:
我正在尝试生成 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)}
【问题讨论】:
标签: python-3.x machine-learning scikit-learn decision-tree grid-search