1. 程式人生 > >KNN演算法身高、體重、鞋號預測男女性別

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)