python - RandomizedSearchCV に分布または None 値から選択するように指示する方法は?

Question

max_depthの最適なパラメータを見つけようとしているとしましょうRandomForestClassifier。使用してい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

今このコードを実行すると、次のエラーが発生します。

Answer 1

また、ドキュメントから：「リストが指定されている場合、それは均一にサンプリングされます。」これを使って：

'max_depth': list(range(1, 100)) + [None]