18 lines
296 B
Python
18 lines
296 B
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: UTF-8 -*-
|
||
|
|
|
||
|
|
from sklearn import neighbors
|
||
|
|
from sklearn import datasets
|
||
|
|
|
||
|
|
knn = neighbors.KNeighborsClassifier()
|
||
|
|
|
||
|
|
iris = datasets.load_iris()
|
||
|
|
|
||
|
|
print iris
|
||
|
|
|
||
|
|
knn.fit(iris.data, iris.target)
|
||
|
|
|
||
|
|
predictedLabel = knn.predict([[0.1, 0.2, 0.3, 0.4]])
|
||
|
|
|
||
|
|
print predictedLabel
|