获得在KNN概率为0或1(predict_proba)

问题描述 投票:4回答:1

我是用KNN从sklearn和使用predict_proba预测标签。我在0到1的范围期待值,因为它告诉特定类的概率。但我只得到0和1。

我已经把大的k值,但还没有增益。虽然我只有1000左右200个特征样本和基质主要是稀疏。

谁能告诉我什么可以解决吗?

scikit-learn knn
1个回答
0
投票

究其原因可能是缺乏各种训练和测试数据集的。

如果样品的特征可以在一个特定的类只存在和其功能不在训练集中的其他类的任何样品中存在,则该样品将被预测为属于该类别为100%(1)和0的概率%(0)用于其他类。除此以外;让说你有2类和测试像knn.predict_proba(sample)样本,并期待像[[0.47, 0.53]]结果将在总在任一方式产生1了一些成绩。

如果多数民众赞成的情况下,尝试生成从训练集中不止一个类的对象都有配备自己的测试样品。

© www.soinside.com 2019 - 2024. All rights reserved.