Изменения

Перейти к: навигация, поиск

Обзор библиотек для машинного обучения на Python

1517 байт добавлено, 14:43, 23 января 2019
Нет описания правки
> '''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'''
====Уменьшение размерности====
333
правки

Навигация