【问题标题】:How do i output the probability of a single prediction being correct?我如何输出单个预测正确的概率?
【发布时间】:2019-12-18 06:46:04
【问题描述】:

我在 Titanic 数据集上使用随机森林分类器。我不想输出乘客是否存活,而是输出乘客存活的概率。使用随机森林可以吗?

当前输出为 0 或 1。我希望响应为例如:87% 的生存机会。

【问题讨论】:

    标签: python machine-learning scikit-learn random-forest


    【解决方案1】:

    我猜你正在使用.predict() 方法?相反,您可以使用.predict_proba()

    预测 X 的类别概率。

    https://scikit-learn.org/stable/modules/generated/sklearn.ensemble.RandomForestClassifier.html#sklearn.ensemble.RandomForestClassifier.predict_proba

    【讨论】:

    猜你喜欢
    • 2021-06-27
    • 2020-06-09
    • 2021-09-28
    • 2016-05-23
    • 2019-09-16
    • 2018-09-01
    • 1970-01-01
    • 2015-02-01
    相关资源
    最近更新 更多