Py学习  »  机器学习算法

如何用机器学习算法设置多个班级?

will Park • 5 年前 • 656 次点击  

我使用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()
Python社区是高质量的Python/Django开发社区
本文地址:http://www.python88.com/topic/30530
 
656 次点击  
文章 [ 2 ]  |  最新文章 5 年前
Vikika
Reply   •   1 楼
Vikika    5 年前

我认为你不需要为xgboost、random forest和mlp做任何额外的事情。对于svc,您可以使用onevsrestClassifier(linearsvc()),然后您只需使用您提到的算法进行培训,并根据预测值对其进行调整,以获得最佳结果。

yatu
Reply   •   2 楼
yatu    5 年前

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