KNN算法身高、体重、鞋号预测男女性别

1.Sklearn代码篇

#预测男女性别
from sklearn.neighbors import KNeighborsClassifier
import numpy as np


neigh=KNeighborsClassifier(n_neighbors=3)
# 训练数据   根据身高  体重  鞋号预测男女性别
X=np.array([[181,80,44],[177,70,43],[160,60,38],[154,54,37],
           [166,65,40],[190,90,47],[175,64,39],[177,70,40],
           [159,55,37],[171,75,42],[181,85,43]])
display(X)
y=['male','male','female','female','male',
  'male','female','female','female','male',
  'male']
display(y)


# 训练模型
neigh.fit(X,y)


# 预测数据
Z=neigh.predict(np.array([[178,77,44],[155,60,39],[178,71,44]]))
print(Z)