关于sklearn Logistic回归的DataConversionWarning

Question

我正在尝试在下面的sklearn中进行物流回归：

from sklearn.linear_model import LogisticRegression

classifier = LogisticRegression(random_state = 0)

mod_data2 = mod_data.copy()

classifier.fit(mod_data2[['prob1_norm', 'prob2_norm']].values.reshape(-1,2), mod_data2['Success'].values.reshape(-1,1))

但是它给了我错误消息：

 DataConversionWarning: A column-vector y was passed when a 1d array was expected. Please change the shape of y to (n_samples, ), for example using ravel().
  y = column_or_1d(y, warn=True)

我尝试在输入数据的末尾使用.ravel（），但是它告诉我尺寸错误。

谢谢

Answer 1

df.squeeze() 应该可以工作。它将数据帧转换为系列，当我使用它时，警告转换消失了

y = mod_data2['成功'].squeeze()