scikit库 fit 函数 训练模型的过程中报错IndexError: too many indices for array

  • Post author:
  • Post category:其他


原本的代码是这样的,进行网格搜索

search1.fit(os_features,os_labels)
search1.grid_scores_, gsearch1.best_params_, gsearch1.best_score_

报错:

IndexError: too many indices for array

百度谷歌之后,发现是传入fit函数的参数格式有错

首先看一下两个数据的shape是否出现(n, )这样的情况

如果没有的话

type(os_features)
type(os_labels)

如果是DataFrame的形式就需要更改一下传入的参数

os_features 改为 os_features.values

os_labels 改为 os_labels[‘keyword’].values

Keyword的值自己查看下自己的label数据 ,就是列id



版权声明:本文为junloin原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。