Как установить несколько классов с алгоритмом машинного обучения?

Я использую XGboost, Randomforest(sklearn), SVM(sklearn) и MLPclassifier(sklearn) в качестве классификатора. И я хочу установить эти модели для класса мульти этикетки. Как я могу установить?

import xgboost as xgb
from sklearn.svm import SVC
from sklearn.neural_network import MLPClassifier
from sklearn.ensemble import RandomForestClassifier

xgb.XGBClassifier()
SVC()
MLPClassifier()
RandomForestClassifier()

2 ответа

Решение

Ни один из упомянутых вами алгоритмов не ограничивается проблемами двоичной классификации. Они могут использоваться для задач мультиклассификации так же, как вы делали бы для двоичной классификации, вызывая model.fit(x_train,y_train),

Я думаю, что вам не нужно ничего делать для XGboost, Random Forest и MLP. Для SVC вы можете использовать OneVsRestClassifier(LinearSVC()). Тогда вам просто нужно потренироваться с упомянутыми вами алгоритмами и настроить его на основе предикторов, чтобы получить наилучшие результаты.

Другие вопросы по тегам