pip install scikit-learn
python
from sklearn import datasets
from sklearn import svm
iris = datasets.load_iris()
X = iris.data
y = iris.target
clf = svm.SVC()
clf.fit(X, y)
new_data = [[5.0, 3.6, 1.3, 0.25], [6.0, 3.4, 4.1, 1.3]]
predicted = clf.predict(new_data)
for i, data in enumerate(new_data):
print('Predicted class for {0}: {1}'.format(data, iris.target_names[predicted[i]]))
