# -*- 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
最新推荐文章于 2023-07-27 11:38:43 发布