机器学习分类

Question

我的目标范围是1到5。是否有办法强制仅在此范围内进行预测？

不管使用哪种模型，有时我都会得到负值和大于5的值。

Answer 1

您可以使用支持多类分类的模型，例如Softmax Regression。该算法是Logistic回归的推广，可以对N个目标进行分类，其中N> 1。

模型的硬预测可以是：

1 2 3 4 5
0 0 0 1 0

这意味着预测为4

或者它可以是一个软预测：

1   2   3   4   5
0.1 0.1 0.6 0.1 0.1

这是概率，然后您可以知道您的模型有多自信。

Scikit-learn通过指定参数multi_class="multinomial"在Logistic regression算法本身中实现Softmax回归