首页 > 解决方案 > 无法训练或测试数据

问题描述

我有这位先生在“ https://github.com/venky14/Machine-Learning-with-Iris-Dataset/blob/master/Machine%20Learning%20with%20Iris%20Dataset.ipynb ”中提到的代码

将数据拆分为训练和测试后,我无法将特征用于训练和测试数据。在 [92] 处抛出错误。

它给了我错误“KeyError:“['A''B''C''D''E''F''H''I'] 不在索引中”

下面是我的 CSV 文件的样子在此处输入图像描述

标签: pythonclassificationsvmcsv

解决方案


您似乎将列名称为索引。请提供示例代码,因为引用的 ipynb 似乎是正确的。可能你正在寻找这个:

import pandas as pd
df = pd.read_csv('sample-table.csv')
df_selected_columns = df[['A', 'B', 'C', 'D', 'E', 'F', 'H', 'I']]
np_ndarray = df_selected_columns.values

推荐阅读