【发布时间】:2016-03-04 21:15:18
【问题描述】:
我正在使用 Scikit-learn 在 Python 中实现多项逻辑回归模型。这是我的代码:
X = pd.concat([each for each in feature_cols], axis=1)
y = train[["<5", "5-6", "6-7", "7-8", "8-9", "9-10"]]
lm = LogisticRegression(multi_class='multinomial', solver='lbfgs')
lm.fit(X, y)
但是,当它尝试执行最后一行代码时,我得到了ValueError: bad input shape (50184, 6)。
X 是一个 DataFrame,有 50184 行,7 列。 y 也有 50184 行,但有 6 列。
我最终想预测结果落在哪个区间(
【问题讨论】:
-
文档要求输入 Y 的向量。也许您应该尝试将训练变量编码为不同的值而不是虚拟变量?
-
@extremist Y的形状需要是(50184,1)
标签: python pandas machine-learning scikit-learn logistic-regression