我的目标范围是1到5。有什么方法可以强制模型仅在这个范围内进行预测吗?
无论我使用哪种模型,有时会得到负值和大于5的值。
回答:
你可以使用支持多类分类的模型,比如Softmax回归
。这种算法是对Logistic回归的扩展,可以对N个目标进行分类,其中N > 1。
你的模型的硬预测可以是:
1 2 3 4 50 0 0 1 0
这意味着预测结果是4
或者它可以是软预测:
1 2 3 4 50.1 0.1 0.6 0.1 0.1
这是概率分布,你可以据此了解模型的置信度。
Scikit-learn在Logistic回归算法中通过指定参数multi_class="multinomial"
实现了Softmax回归。