sklearn中的网格搜索交叉验证

1 回答

• 8

  为什么不 ？

  我邀请您查看GridsearchCV的文档 .

  示例

  from sklearn.tree import DecisionTreeClassifier
  from sklearn.metrics import roc_auc_score
  
  param_grid = {'max_depth': np.arange(3, 10)}
  
  tree = GridSearchCV(DecisionTreeClassifier(), param_grid)
  
  tree.fit(xtrain, ytrain)
  tree_preds = tree.predict_proba(xtest)[:, 1]
  tree_performance = roc_auc_score(ytest, tree_preds)
  
  print 'DecisionTree: Area under the ROC curve = {}'.format(tree_performance)
  

  并提取最佳参数：

  tree.best_params_
  Out[1]: {'max_depth': 5}
  

  回复于 2024-04-30T12:10:47+08:00