gpt4 book ai didi

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

转载 作者:太空狗 更新时间:2023-10-30 00:43:37 28 4
gpt4 key购买 nike

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

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

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

我尝试了以下代码,但这只为每个样本提供了一个分类。

from sklearn.multiclass import OneVsRestClassifier
clf = OneVsRestClassifier(DecisionTreeClassifier())
clf.fit(X_train, y_train)
pred = clf.predict(X_test)

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

最佳答案

您可以通过简单地删除 OneVsRestClassifer 并使用 predict_proba 来做到这一点DecisionTreeClassifier 的方法.您可以执行以下操作:

clf = DecisionTreeClassifier()
clf.fit(X_train, y_train)
pred = clf.predict_proba(X_test)

这将为您提供 7 个可能类别中每个类别的概率。

希望对您有所帮助!

关于python - Sklearn - 如何预测所有目标标签的概率,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/38403951/

28 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com