从已加载的ARFF文件中创建火车和测试变量

Question

我想执行多标签分类。A具有我加载的ARFF格式的数据集。但是，我现在没有如何将导入数据转换为x和y向量以应用Sklearn/train_testrongplit。

我如何获得x和y？

data, meta = scipy.io.arff.loadarff('../yeast-train.arff')
df = pd.DataFrame(data)
#Get X, y
X, y = ??? <---
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2)

Answer 1

好。它是一个多标签数据，其中列中的功能在Att1, Att2, Att3.... Att20中，目标位于列中Class1, Class2, .... Class14。

。

因此，您需要使用这些列来获取X和Y。这样做：

# Fill the .... with all other column names
feature_cols = ['Att1', 'Att2', 'Att3', 'Att4', 'Att5' ....   'Att20']
target_cols = ['Class1', 'Class2', 'Class3', 'Class4', ....   'Class14']
X, y = df[feature_cols], df[target_cols]