0

我想将 BayesSearchCV 与某些实数和整数一起用于参数学习率和深度。下面的代码运行 0.001 和 0.01 之间的实数以及 1 和 3 之间的深度整数。但我只需要搜索 0.001 和 0.01 之间的实数，而不是 1 和 3 的深度整数。我该怎么做？请帮忙。谢谢你。

     param_grid= {
    'learning_rate': [0.001,0.01],
     "depth": [1,3] 
     }
  
     #model
     clf=CatBoostClassifier(
     iterations=10,
     cat_features=cat_features,
     verbose=100,
     early_stopping_rounds=20
     )

     grid_search_bayes=BayesSearchCV(clf,param_grid,cv=5,scoring='roc_auc')
     grid_search_results_bayes=grid_search_bayes.fit(X,y,cat_features=cat_features)
     grid_search_results_bayes.best_estimator_.get_params()
4

0 回答 0