Печать соответствующего значения массива - PullRequest
0 голосов
/ 21 марта 2020

Я пытаюсь напечатать количество градусов модели SVM, которая предлагает самую высокую степень точности. Я не могу использовать max (degree_matrix_name), потому что это дает мне наибольшее количество проверенных степеней.

x_axis_knn_array = []
y_axis_knn_array = []
from sklearn.neighbors import KNeighborsClassifier
import matplotlib as matplotlib 
import matplotlib.pyplot as plt
for b in range (6, 71):
    knnClassObj = KNeighborsClassifier(n_neighbors=b, p=2, metric='minkowski')
    knnClassObj.fit(X_train, y_train)
    knn_y_pred = knnClassObj.predict(X_test)
    y_axis_knn_array.append(metrics.accuracy_score(y_test, knn_y_pred) * 100)
    x_axis_knn_array.append(b)
print("The highest accuracy is", max(y_axis_knn_array), "%", "at", #**[I WANT TO HAVE THE NUMBER OF DEGREES CORRESPONDING TO max_y_axis_knn_array HERE]**)

1 Ответ

0 голосов
/ 21 марта 2020

Вы можете индексировать x_axis_knn_array, используя местоположение максимальной точности в массиве точности

x_axis_knn_array[y_axis_knn_array.index(max(y_axis_knn_array))]
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...