#python #python_3x #машинное_обучение #scikit_learn
from sklearn import datasets iris=datasets.load_iris() x=iris.data y=iris.target from sklearn.model_selection import train_test_split x_train,x_test,y_train,y_test=train_test_split(x,y,test_size=.5) from sklearn import neighbors classifier=neighbors.KNeighborsClassifier() classifier.fit(x_train,y_train) predictions=classifier.predict(x_test) from sklearn.metrics import accuracy_score print(accuracy_score(y_test,predictions)) # 0.9333333333333333 Каким образом мне можно проверить модель, задав ему X - ы, что бы он выдал мне предикт по цветку:)
Ответы
Ответ 1
Вы уже проверили модель в данной строке: predictions=classifier.predict(x_test) и получили массив предсказанных цветков: In [308]: predictions Out[308]: array([1, 1, 1, 0, 2, 0, 0, 0, 0, 1, 0, 1, 2, 1, 1, 1, 2, 0, 2, 1, 1, 2, 1, 1, 1, 1, 0, 0, 2, 1, 0, 2, 2, 0, 2, 2, 2, 2, 1, 2, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 2, 2, 1, 1, 2, 1, 2, 0, 0, 1, 2, 2, 0, 0, 0, 2, 0, 0, 2, 0, 0, 0, 2, 2, 0]) чтобы получить соответствующие имена цветков: In [305]: X_test_names = np.take(iris.target_names, predictions) In [306]: X_test_names Out[306]: array(['versicolor', 'versicolor', 'versicolor', 'setosa', 'virginica', 'setosa', 'setosa', 'setosa', 'setosa', 'versicolor', 'setosa', 'versicolor', 'virginica', 'versicolor', 'versicolor', 'versicolor', 'virginica', 'setosa', 'virginica', 'versicolor', 'versicolor', 'virginica', 'versicolor', 'versicolor', 'versicolor', 'versicolor', 'setosa', 'setosa', 'virginica', 'versicolor', 'setosa', 'virginica', 'virginica', 'setosa', 'virginica', 'virginica', 'virginica', 'virginica', 'versicolor', 'virginica', 'setosa', 'setosa', 'setosa', 'versicolor', 'versicolor', 'versicolor', 'versicolor', 'versicolor', 'versicolor', 'versicolor', 'virginica', 'virginica', 'versicolor', 'versicolor', 'virginica', 'versicolor', 'virginica', 'setosa', 'setosa', 'versicolor', 'virginica', 'virginica', 'setosa', 'setosa', 'setosa', 'virginica', 'setosa', 'setosa', 'virginica', 'setosa', 'setosa', 'setosa', 'virginica', 'virginica', 'setosa'], dtype='
Комментариев нет:
Отправить комментарий