如果你看看 文档 load_boston() 你会看到它返回一个 Bunch 宾语。如果您在Spyder的变量资源管理器中检查该对象,您可以看到它包含描述,实际数据(您可以进行预测的功能),每个功能的标签以及包含您正在尝试的值的目标向量预测。
load_boston()
Bunch
如果您只想获取数据部分(用于预测的数据要素),则可以运行以下命令:
train_set, test_set = train_test_split(housing_data.data, test_size = 0.2, random_state = 42)
或者,您可以使用以下内容为X和y(要素和目标)创建训练和测试集:
X_train, X_test, y_train, y_test = train_test_split(housing_data.data, housing_data.target, test_size = 0.2, random_state = 42)
这会产生以下变量集:
的 编辑: 强> 如果你用load调用load_boston() return_X_y = True 参数,它返回一个元组 (data, target) ,允许你做以下,这可以说是更优雅:
return_X_y = True
(data, target)
X, y = load_boston(return_X_y = True) X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.2, random_state = 42)