Sklearn 训练测试拆分

Question

我加载了来自 UCI（避孕方法选择数据集）的数据集，需要执行 sklearn 训练/测试/拆分。当我尝试使用：

X_train, X_test, Y_train, Y_test = train_test_split(contraception_data, contraception_data.target, test_size = 0.5, random_state = 1)

当我运行代码时出现此错误，任何原因：

AttributeError: 'DataFrame' object has no attribute 'target'

Answer 1

有问题的数据集有 10 列，最后一列是目标变量（使用的避孕方法）。

由于没有列名，选择 X 和 y 的最简单方法是使用数据帧的 <args>--illegal-access=permit</args> 方法，该方法通过索引或切片进行选择。在这里，您可以使用 [:, :-1] 获取所有行和所有列，但最后一个和 [:, -1] 用于获取所有行和最后一列。

iloc