【发布时间】:2017-09-19 15:37:43
【问题描述】:
假设我们试图找到RandomForestClassifier 的最佳max_depth 参数。我们正在使用RandomizedSearchCV:
from scipy.stats import randint as sp_randint
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import RandomizedSearchCV
rf_params = { # Is this somehow possible?
'max_depth': [sp_randint(1, 100), None],
}
n_iter = 10
random_search = RandomizedSearchCV(RandomForestClassifier(),
verbose=50,
param_distributions=rf_params,
n_iter=n_iter,
n_jobs=-1,
scoring='f1_micro')
random_search.fit(X_train, y_train)
是否可以告诉RandomizedSearchCV 从指定分布sp_randint(1, 100) 中进行选择或将参数设置为None,这将(如文档中所示):"...展开节点,直到所有叶子都是纯的或者直到所有叶子包含少于 min_samples_split 个样本..."?
当我现在运行这段代码时,我会得到这个错误:
【问题讨论】:
标签: python scipy scikit-learn random-forest hyperparameters