美文网首页
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