如何通过网格搜索绘制最佳超参数的 AUC



我正在使用以下代码来获得最佳超参数,该参数将通过网格搜索给出最大的AUC值。 但我无法绘制它。

model = KNeighborsClassifier()
#Hyper Parameters Set
params = {'n_neighbors':[5,10,15,20,25,30,35],
'leaf_size':[1,2,3,4],
'weights':['uniform', 'distance'],
'algorithm':['brute']
}
#Making models with hyper parameters sets
model1_tfidf = GridSearchCV(model, param_grid=params)
model1_tfidf.fit(final_X_train_tfidf,y_train)
print("Best Hyper Parameters:n",model1.best_params_)

GridsearchCV 只返回 CV 分数,但不会绘制 roc 曲线。

遵循以下堆栈溢出链接: 如何在 Python 中绘制 ROC 曲线

最新更新