如何使cross_val_score与pandas DataFrame一起工作

Question

我刚建立了自己的分类器，我想用这个函数测试它： scores = cross_val_score(model, x, y, cv=10)。

model是我自己的分类器，它使用pandas DataFrame（x和y是数据帧的类型）。

问题是sklearn.model_selection.cross_val_score不能与pandas DataFrame一起使用，我怎样才能使它工作？

Answer 1

试试这个：

scores = cross_val_score(model, x.values, y.values, cv=10)

cross_val_score 功能需要列表，而不是DataFrame，因此您只需将数据帧转换为列表即可。