相关疑难解决方法(0)

Sklearn-如何预测所有目标标签的概率

我有一个带有目标变量的数据集,该变量可以具有7个不同的标签。我的训练集中的每个样本都只有一个目标变量标签。

对于每个样本,我想计算每个目标标签的概率。所以我的预测将由每行7个概率组成。

在sklearn网站上,我读到了有关多标签分类的信息,但这似乎不是我想要的。

我尝试了以下代码,但是每个样本只能给我一个分类。

from sklearn.multiclass import OneVsRestClassifier
clf = OneVsRestClassifier(DecisionTreeClassifier())
clf.fit(X_train, y_train)
pred = clf.predict(X_test)
Run Code Online (Sandbox Code Playgroud)

有人对此有建议吗?谢谢!

python scikit-learn multilabel-classification

5
推荐指数
1
解决办法
9786
查看次数