Sklearn多类分类类顺序

时间:2018-04-20 21:39:56

标签: python pandas scikit-learn

我有三个类[-1,0,1],我正在对它们运行多类逻辑回归。当我运行logreg.predict_proba(x)时,它返回一个数组[.25,.5,。25]这是否意味着位置0是-1级,位置1是0级,位置2是1级?换句话说,逻辑回归如何将类映射到输出列?它是通过数字顺序完成的吗?或者基于它看到的第一堂课?

1 个答案:

答案 0 :(得分:3)

您可以使用逻辑回归分类器的classes属性验证类的顺序。例如,如果分类器名为logreg,那么

logreg.classess_

将显示课程的顺序。

http://scikit-learn.org/stable/modules/generated/sklearn.linear_model.LogisticRegression.html#sklearn.linear_model.LogisticRegression.predict_proba