# -*- coding:utf-8 -*-
import numpy as np
import pandas as pd
from sklearn.neighbors import KNeighborsClassifier
model = KNeighborsClassifier()
origin_data = pd.read_csv('https://archive.ics.uci.edu/ml/machine-learning-databases/iris/iris.data',names=['sepal_length','sepal_width','petal_length','petal_width','class_name'])
norm_data = origin_data.apply(lambda v : v if v.name == 'class_name' else (v - np.min(v))/(np.max(v) - np.min(v)))
train_data = norm_data.sample(frac=0.7)
test_data = norm_data.drop(train_data.index)
model.fit(X=train_data.iloc[:,:-1],y=np.array(train_data.iloc[:,-1:]).astype(str).ravel())
accuracy = model.score(X=test_data.iloc[:,:-1],y=np.array(test_data.iloc[:,-1:]).astype(str).ravel())
print(accuracy)
just_try = model.predict(X=test_data.iloc[:,:-1])
print(just_try)
机器学习之Hello World kNN
猜你喜欢
转载自blog.csdn.net/mscf/article/details/76796714
今日推荐
周排行