333
правки
Изменения
Нет описания правки
> '''Score for naive Bayes: 0.774'''
'''Score for SVM: 0.824'''
====Метод опорных векторов (SVM)====
Основная статья: [[Метод опорных векторов (SVM)]].
Загрузка датасета:
'''from''' sklearn '''import''' datasets
iris = datasets.load_iris()
Разбиение датасета на тестовый и тренировочный:
'''from''' sklearn.model_selection '''import''' train_test_split
x_train, x_test, y_train, y_test = train_test_split(iris.data, iris.target, test_size=<font color="blue">0.25</font>, random_state=<font color="blue">0</font>)
Построение и обучение модели:
clf = svm.SVC(kernel=<font color="red">'linear'</font>, C=<font color="blue">1.0</font>)
clf.fit(x_train, y_train)
predictions = clf.predict(x_test)
Оценка алгоритма:
'''from''' sklearn.metrics '''import''' classification_report, confusion_matrix
print(confusion_matrix(y_test,predictions))
print(classification_report(y_test,predictions))
> '''[[13 0 0]'''
'''[ 0 15 1]'''
'''[ 0 0 9]]'''
'''precision recall f1-score support'''
'''0 1.00 1.00 1.00 13'''
'''1 1.00 0.94 0.97 16'''
'''2 0.90 1.00 0.95 9'''
'''micro avg 0.97 0.97 0.97 38'''
'''macro avg 0.97 0.98 0.97 38'''
'''weighted avg 0.98 0.97 0.97 38'''
====Уменьшение размерности====