Python - 分类Lasso sklearn - 如何预测类

时间:2016-04-18 14:46:27

标签: python scikit-learn classification

以下示例: http://scikit-learn.org/stable/modules/generated/sklearn.linear_model.Lasso.html

from sklearn import linear_model
clf = linear_model.Lasso(alpha=0.1)
clf.fit([[0,0], [1, 1], [2, 2]], [0, 1, 2])

clf.predict(np.array([0,0]).reshape(1,-1))
Out[13]: array([ 0.15])

我可以将预测作为分类而不是回归。换句话说,当我给它一个输入时,我想要一个绝对的输出。

2 个答案:

答案 0 :(得分:2)

LogisticRegressionpenalty='l1' 一起使用。
它本质上是套索回归,但附加了一层将类的分数转换为“获胜”类输出标签。
正则化强度由C定义,它是alpha的INVERSE,由Lasso使用。
Scikit-learn 对线性模型有一个非常好的简要概述:
https://scikit-learn.org/stable/modules/linear_model.html

答案 1 :(得分:0)

您可以使用套索对预测结果进行降序排序,因此前50%为1,最后为0。