无法训练或测试数据

时间:2018-12-25 13:46:50

标签: python classification svm csv

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

将数据分为训练和测试之后,我无法使用训练和测试数据的功能。In[92]处抛出错误。

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

下面是我的CSV文件的外观图片,enter image description here

1 个答案:

答案 0 :(得分:1)

似乎您将列名称为索引。 请提供示例代码,因为所引用的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