你好,我使用决策树模型编写了机器学习程序。我创建了一个Web应用程序,用户可以输入数据,Web应用程序会通过Flask API调用模型,然后在Web应用程序上显示结果,但我的结果只有是/否。是否有可能让结果显示这个输入是/否的百分比,例如“是 76%”?
回答:
你可以使用predict_proba(),因为它会返回输入样本X的类别概率。
probs = clf.predict_proba(X)# probs将输出每个类别的预测概率。
你可以在文档中以及其他类似问题的帖子中了解更多信息。
https://scikit-learn.org/stable/modules/generated/sklearn.tree.DecisionTreeClassifier.html