如何创建scikit学习数据集

Question

我有一个数组，其中第一列是类（整数形式），其余列是特征。

像这样的SG

1,0,34,23,2
0,0,21,11,0
3,11,2,11,1

我

怎样才能把它变成一个与scikit兼容的数据集，这样我就可以像这样称呼sg。我的数据集 = datasets.load_mydataset（）？

Answer 1

你可以简单地使用熊猫，例如，如果你已经将数据集复制到数据集.csv文件。只需适当地标记 csv 文件中的列即可。

In [1]: import pandas as pd
In [2]: df = pd.read_csv('temp.csv')
In [3]: df
Out[3]: 
   Label  f1  f2  f3  f4
0      1   0  34  23   2
1      0   0  21  11   0
2      3  11   2  11   1
In [4]: y_train= df['Label']
In [5]: x_train = df.drop('Label', axis=1)
In [6]: x_train
Out[6]: 
   f1  f2  f3  f4
0   0  34  23   2
1   0  21  11   0
2  11   2  11   1
In [7]: y_train
Out[7]: 
0    1
1    0
2    3