arthur/handson-ml3
1
0
Fork 0
mirror of https://github.com/ArthurDanjou/handson-ml3.git synced 2026-01-14 12:14:36 +01:00
Code Issues Packages Projects Releases Wiki Activity

Replace housing.loc[train_index] with housing.iloc[train_index]

Browse Source
This commit is contained in:
Aurélien Geron
2022-09-05 17:07:10 +12:00
parent b2103e953a
commit 2855982461
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
02_end_to_end_machine_learning_project.ipynb
View File

@@ -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])"
]
},