【发布时间】:2019-12-18 06:46:04
【问题描述】:
我在 Titanic 数据集上使用随机森林分类器。我不想输出乘客是否存活,而是输出乘客存活的概率。使用随机森林可以吗?
当前输出为 0 或 1。我希望响应为例如:87% 的生存机会。
【问题讨论】:
标签: python machine-learning scikit-learn random-forest
我在 Titanic 数据集上使用随机森林分类器。我不想输出乘客是否存活,而是输出乘客存活的概率。使用随机森林可以吗?
当前输出为 0 或 1。我希望响应为例如:87% 的生存机会。
【问题讨论】:
标签: python machine-learning scikit-learn random-forest
我猜你正在使用.predict() 方法?相反,您可以使用.predict_proba()
预测 X 的类别概率。
【讨论】: