我在泰坦尼克数据集上使用随机森林分类器。我不想输出乘客是否存活,而是想输出乘客存活的概率。使用随机森林可以实现吗?
目前的输出结果是0或1。我希望得到的回应是,例如:存活的概率为87%。
回答:
我想你使用的是.predict()
方法吧?你可以改用.predict_proba()
方法。
为X预测类别概率。
我在泰坦尼克数据集上使用随机森林分类器。我不想输出乘客是否存活,而是想输出乘客存活的概率。使用随机森林可以实现吗?
目前的输出结果是0或1。我希望得到的回应是,例如:存活的概率为87%。
回答:
我想你使用的是.predict()
方法吧?你可以改用.predict_proba()
方法。
为X预测类别概率。