머신러닝2일차 (연습) 몸무게와 키를 보고 여자인지 남자인지 확인해보자!

#  몸무게와 키를 보고 여자인지 남자인지 확인해보자!

#  남자 5명  여자 5명

사람키 = [180, 190, 175, 181, 182, 168, 170, 158, 160, 150]

사람무게 = [75, 90, 68, 72, 73, 50, 58, 52, 48, 45]

사람특성 = [[키,무게]for 키, 무게 in zip(사람키,사람무게)]

사람정답 = [1, 1, 1, 1, 1, 0, 0, 0, 0, 0 ]

from sklearn.neighbors import KNeighborsClassifier

# 학습 시에 특성은 2차원 리스트, 타겟은 1차원 리스트이다!

최근접이웃모델 = KNeighborsClassifier()

최근접이웃모델.fit(사람특성, 사람정답)

# fit() 데이터 2차원 리스트여야한다

# target 데이터 1차원

# 예측 시에도 특성은 2차원 리스트로 전달해야 예측을 제대로 수행한다!

print(최근접이웃모델.predict([[158, 52]]))  #  [0]

 

사람정답으로 확인한결과  사람키 158, 사람몸무게는 52  여성인것을 확인할수 있다.