[머신러닝] Pandas 와 metrics 를 이용해서 XOR 연산을 더 간단히

Pandas 모듈의 DataFrame 기능을 사용하여 구현했습니다.

데이터 학습과 예측은 이전과 같습니다. 마지막으로 정답률 구하는 것은

metrics.accuracy_score() 함수를 이용하여 정답률을 쉽게 구하였습니다. 

매개변수에는 실제 답과 예측 결과 배열을 전달하면 됩니다.

import pandas as pd
from sklearn import svm, metrics
 
# XOR 연산
xor_input = [
    [0,0,0],
    [0,1,1],
    [1,0,1],
    [1,1,0]
]
# 입력을 학습 전용 데이터와 테스트 전용 데이터로 분류
xor_df = pd.DataFrame(xor_input)
xor_data = xor_df.ix[:,0:1] # 데이터
xor_label = xor_df.ix[:,2] # 레이블
 
# 데이터 학습과 예측하기
clf = svm.SVC()
clf.fit(xor_data, xor_label)
pre = clf.predict(xor_data)
 
# 정답률 구하기
ac_score = metrics.accuracy_score(xor_label, pre)
print("정답률 : ", ac_score)