언어/python

machine "svm"

파아랑새 2018. 7. 9. 19:05
from sklearn import svm
import random as rd

train_data = list()
result_data = list()

for _ in range(100):
train_data.append([rd.uniform(1., 1.9) for _ in range(2)])
result_data.append(1)

for _ in range(100):
train_data.append([rd.uniform(3., 3.9) for _ in range(2)])
result_data.append(3)

print (train_data)
# 객체 생성 _________________
clf = svm.SVC()
clf.fit(train_data, result_data)
print (clf.predict([[2.1, 3.1]]))