Add TP1 exo2 question 2 implementation.
[TP_AA.git] / TP1 / exo2 / TP1prog2.py
CommitLineData
7761b833
JB
1#!/usr/bin/env python3
2
3# -*- coding: utf-8 -*-
4from sklearn import neighbors
5from sklearn.datasets import load_iris
6irisData = load_iris()
7
8X = irisData.data
9Y = irisData.target
10
7761b833
JB
11nb_voisins = 15
12clf = neighbors.KNeighborsClassifier(nb_voisins)
13clf.fit(X, Y)
14print("kNN prediction on [5.4, 3.2, 1.6, 0.4]:")
15print(clf.predict([[5.4, 3.2, 1.6, 0.4]]))
16print("kNN probability prediction on [5.4, 3.2, 1.6, 0.4]:")
17print(clf.predict_proba([[5.4, 3.2, 1.6, 0.4]]))
18print("kNN score on Iris data:")
19print(clf.score(X, Y))
20print("kNN prediction error(s) on Iris data:")
21Z = clf.predict(X)
22print(X[Z != Y])