From 28559824611de0cbfc75ff27f3bb9978be003611 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Aur=C3=A9lien=20Geron?= Date: Mon, 5 Sep 2022 17:07:10 +1200 Subject: [PATCH] Replace housing.loc[train_index] with housing.iloc[train_index] --- 02_end_to_end_machine_learning_project.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/02_end_to_end_machine_learning_project.ipynb b/02_end_to_end_machine_learning_project.ipynb index b0f72e7..e1ee729 100644 --- a/02_end_to_end_machine_learning_project.ipynb +++ b/02_end_to_end_machine_learning_project.ipynb @@ -835,8 +835,8 @@ "splitter = StratifiedShuffleSplit(n_splits=10, test_size=0.2, random_state=42)\n", "strat_splits = []\n", "for train_index, test_index in splitter.split(housing, housing[\"income_cat\"]):\n", - " strat_train_set_n = housing.loc[train_index]\n", - " strat_test_set_n = housing.loc[test_index]\n", + " strat_train_set_n = housing.iloc[train_index]\n", + " strat_test_set_n = housing.iloc[test_index]\n", " strat_splits.append([strat_train_set_n, strat_test_set_n])" ] },