인공지능 머신러닝 - 최근접 이웃 모델(K-Nearest Neighbor =K-NN)

반응형

# 최근접 이웃 (K-Nearest Neighbor =K-NN)

  • 특정 공간내 가장 가까운 데이터의 갯수로 인접한 카테고리로 분류하는 것

 

# 최근접 이웃 모델(K-NN) 이해하기

  • 새로운 데이터가 추가 될 경우, 분류를 어떻게 할 것인가?

 

  • 가장 가까운 데이터들의 갯수를 확인

 

  • 인접 데이터의 갯수가 많은 카테고리로 분류

 

# 최근접이웃 라이브러리 호출

import matplotlib.pyplot as plt
# 표 작성을 위한 라이브러리

import pandas
# 데이터프레임 사용을 위한 라이브러리

from sklearn.preprocessing import StandardScaler
# 피처 스케일링 표준화 사용을 위한 라이브러리

from sklearn.model_selection import train_test_split
# 인공지능 학습을 위한 라이브러리

from sklearn.neighbors import KNeighborsClassifier
# 최근접 이웃 인공지능 모델링을 위한 라이브러리

from sklearn.metrics import confusion_matrix, accuracy_score
# 인공지능 성능을 확인하기 위한 라이브러리

 

# 최근접 이웃(K-NN) 모델로 인공지능 학습하기

  • 나이와 연봉을 분석해서 물건 구매 여부를 예측
  • 예시에 쓰일 데이터프레임

 

 

 

 

1. 피처 스케일링

  • 우선 NaN값이 존재하는지 확인, 존재하면 NaN처리

 

  • 독립 변수와 종속 변수 지정, 데이터(X, y) 분리
    • 독립 변수(X) : 나이(Age) / 연봉(EstimatedSalary)
    • 종속 변수(y) : 구매횟수(Purchased)

 

  • 데이터 표준화 (정규화 해도 됨)
    • from sklearn.preprocessing import StandardScaler 호출

 

2. 데이터(학습용/테스트용) 분리

  • from sklearn.model_selection import train_test_split 호출

 

3. 인공지능 최근접 이웃 모델링

  • from sklearn.neighbors import KNeighborsClassifier 호출
  • n_neighbors=n : 근접한 데이터의 갯수

 

4. 인공지능 성능 테스트

 

5. 인공지능 성능 확인

  • from sklearn.metrics import confusion_matrix, accuracy_score 호출
  • 정확도 확인

정확도는 90%

 

6. 인공지능 학습 결과 시각화

from matplotlib.colors import ListedColormap
X_set, y_set = X_test, y_test
X1, X2 = np.meshgrid(np.arange(start = X_set[:, 0].min() - 1, 
                               stop = X_set[:, 0].max() + 1, step = 0.01),
                     np.arange(start = X_set[:, 1].min() - 1, 
                               stop = X_set[:, 1].max() + 1, step = 0.01))
plt.figure(figsize=[10,7])
plt.contourf(X1, X2, classifier.predict(
            np.array([X1.ravel(), X2.ravel()]).T).reshape(X1.shape),
             alpha = 0.75, cmap = ListedColormap(('red', 'green')))
plt.xlim(X1.min(), X1.max())
plt.ylim(X2.min(), X2.max())
for i, j in enumerate(np.unique(y_set)):
    plt.scatter(X_set[y_set == j, 0], X_set[y_set == j, 1],
                c = ListedColormap(('red', 'green'))(i), label = j)
plt.title('Classifier (Test set)')
plt.legend()
plt.show()

반응형