Skip to content

Commit 3a9b423

Browse files
student-ChestaVashishthastudent-ChestaVashishtha
authored andcommitted
Add cross-validation diagram to GridSearchCV notebook
1 parent 5231792 commit 3a9b423

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

python_scripts/parameter_tuning_grid_search.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -124,9 +124,13 @@
124124
# Let's see how to use the `GridSearchCV` estimator for doing such search. Since
125125
# the grid-search is costly, we only explore the combination learning-rate and
126126
# the maximum number of nodes.
127+
### Cross-Validation Visualization
128+
![Grid Search CV](../figures/cross_validation_train_test_diagram.png)
129+
130+
#This diagram illustrates how GridSearchCV uses cross-validation to split the dataset during hyperparameter tuning.
127131

128-
# %%
129132
# %%time
133+
130134
from sklearn.model_selection import GridSearchCV
131135

132136
param_grid = {

0 commit comments

Comments
 (0)