머신러닝 패키지 정리 (machine learning package) - 1

머신러닝 패키지 정리 - 1

matplotlib(맷플롭립)

• scatter() : 그래프에서 산점도 그리기
• show() : 아래와 같이 그래프를 보여줌

import matplotlib.pyplot as plt
plt.scatter(x좌표, y좌표)
plt.show()

---

scikit-learn(사이킷런)

scikit-learn을 사용하기 위해서는 2차원 리스트가 필요하다.

• zip() : 나열된 리스트에서 원소를 하나씩 꺼내 줌

xy_data = [[l, w] for l, w in zip(x_data, y_data)]

위와 같은 방법으로 1차원 리스트들을 대칭되게 2차원 리스트로 변경

 

• KNeighborsClassifier()

from sklearn.neighbors import KNeighborsClassifier

※ from ~import 사용 이유 : 만약 import만 사용한다면 아래와 같이 클래스 이름을 다 적어줘야 함

import sklearn
model = sklearn.neighbors.KNeighborsClassifier()

 

• 중요 메서드 : fit(), score() predict()

fit(특징데이터, target 값) : 주어진 데이터로 모델을 훈련시키는 메서드

score(특징데이터, target 값) : 모델을 평가하는 메서드

predict(리스트) : 새로운 데이터의 정답을 예측하는 메서드, fit()에서 특징데이터처럼 리스트로 입력