Add TP1 exo2 implementation (question 1).
[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
11colors = ["red", "green", "blue"]
12
13nb_voisins = 15
14clf = neighbors.KNeighborsClassifier(nb_voisins)
15clf.fit(X, Y)
16print("kNN prediction on [5.4, 3.2, 1.6, 0.4]:")
17print(clf.predict([[5.4, 3.2, 1.6, 0.4]]))
18print("kNN probability prediction on [5.4, 3.2, 1.6, 0.4]:")
19print(clf.predict_proba([[5.4, 3.2, 1.6, 0.4]]))
20print("kNN score on Iris data:")
21print(clf.score(X, Y))
22print("kNN prediction error(s) on Iris data:")
23Z = clf.predict(X)
24print(X[Z != Y])