弄清楚如何在sklearn中编写逻辑回归模型

Question

所以我试图用sklearn编写一个模型。我的训练集的维度为xtrain（num_samples，65000,2），维度为ytrain（num_samples）。

我正在使用

clf = LogisticRegression(random_state=0, solver='lbfgs',
                          multi_class='multinomial').fit(xtrain, ytrain)

但是出现错误：

  

找到具有暗3的数组。估计量应为<= 2。

任何人都可以建议我如何解决此问题。

谢谢

Answer 1

该错误表明您正在传递3d数组，而应该传递2d数组。

也许我们可以看到您的数据，那么我们可以为您提供更多帮助。

这应该可以解决您的错误：

samples, x, y = xtrain.shape
new_dataset = xtrain.reshape((samples,x*y))

然后再次适合新数据集