Skip to content

Commit 1ae4724

Browse files
author
ercbk
committed
restored n_iter objects back to original
1 parent 8d83798 commit 1ae4724

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

duration-experiment/raschka/nested-cv-py-raschka.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,9 @@
169169
inner_cv = KFold(n_splits = 2, shuffle = True, random_state = 1)
170170

171171
# Setting this parameter to the size of the grid tells Random Search to use every grid value once
172-
iter_dict = {'Elastic Net': len(elast_params), 'Random Forest': len(rf_params)}
172+
elast_iter = len(elast_params)
173+
rf_iter = len(rf_params)
174+
iter_dict = {'Elastic Net': elast_iter, 'Random Forest': rf_iter}
173175

174176

175177
# Setting up multiple RandomSearchCV objects, 1 for each algorithm

0 commit comments

Comments
 (0)