Sklearn多类分类类别顺序

Dav*_*ein 2 python pandas scikit-learn

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

KRK*_*rov 5

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

logreg.classes_
Run Code Online (Sandbox Code Playgroud)

将揭示课程的顺序。

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