美文网首页
XGBOOST调参时出现A column-vector y wa

XGBOOST调参时出现A column-vector y wa

作者: 丙吉 | 来源:发表于2020-11-04 13:49 被阅读0次

今天 用XGBOOST调参的时候碰到了这个问题:

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().

  return f(**kwargs)

原因是,在训练时的y_train的格式不对。

用ravel转换下数据格式即可运行。

y_train = y_train.values.ravel()

或把label值转成list

 y_train = y_train['label'].tolist()

相关文章

网友评论

      本文标题:XGBOOST调参时出现A column-vector y wa

      本文链接:https://www.haomeiwen.com/subject/meqgvktx.html