Question

我正在使用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()

Answer 1

您提到的这些算法都没有局限于二进制分类问题。通过调用model.fit(x_train,y_train)，可以像处理二进制分类一样将它们用于多重分类问题。

Answer 2

我认为您不需要为XGboost，Random forest和MLP做任何额外的事情。对于SVC，您可以使用OneVsRestClassifier（LinearSVC（））。然后，您只需训练上述算法，并根据预测变量对其进行调整即可获得最佳结果

如何用机器学习算法设置多类？

2 个答案: