如何解决如何从 RandomForestClassifier 输出软标签预测而不是 0 或 1
我正在用二元分类进行 kaggle 比赛。我想输出概率输出(软标签 - 0 和 1 之间的某个值),而不仅仅是 0 和 1。不过不确定如何输出我的预测?使用随机森林分类器
解决方法
使用模型的函数predict_proba(X)
。
此函数预测 X 的类别概率。
文档说:
输入样本的预测类别概率计算为 森林中树木的平均预测类别概率。这 单个树的类概率是样本的分数 叶子中的同一个类。
因此,假设您的 RandomForest 是为 100 种不同的树形成的。如果其中 91 棵树预测样本为 0 类,9 棵树预测其为 1 类,则 predict_proba(X)
的输出将是:
[0.91,0.09]
而如果您使用 predict(X),您将直接获得类别预测(在这种情况下,类别 0,因为 0.91 大于 0.09):
[0] # Which refers to class 0
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。