Как построить AU C для лучших гиперпараметров с помощью поиска по сетке - PullRequest
0 голосов
/ 17 июня 2020

Я использую приведенный ниже код, чтобы получить лучший гиперпараметр, который даст максимальное значение C AU при поиске по сетке. но я не могу это построить.

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_)

1 Ответ

0 голосов
/ 18 июня 2020

GridsearchCV просто возвращает оценки CV, но не строит кривую ro c.

Следуйте приведенной ниже ссылке на переполнение стека: Как построить кривую RO C в Python

...