Как установить несколько классов с алгоритмом машинного обучения?
Я использую 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()). Тогда вам просто нужно потренироваться с упомянутыми вами алгоритмами и настроить его на основе предикторов, чтобы получить наилучшие результаты.