Évaluer des modèles
Deux modèles ont été entraînés et sont disponibles : large_model, qui comporte de nombreux paramètres, et small_model, qui en comporte moins. Les deux modèles ont été entraînés avec train_features et train_labels, qui sont à votre disposition. Un jeu de test distinct, composé de test_features et test_labels, est également disponible.
Votre objectif est d'évaluer la performance relative des modèles et de déterminer si l'un ou l'autre présente des signes de surapprentissage (overfitting). Pour ce faire, vous évaluerez large_model et small_model à la fois sur le jeu d'entraînement et sur le jeu de test. Pour chaque modèle, appliquez la méthode .evaluate(x, y) pour calculer la perte à partir des caractéristiques x et des étiquettes y. Vous comparerez ensuite les quatre valeurs de perte obtenues.
Cette activité fait partie du cours
Introduction à TensorFlow en Python
Instructions de l’exercice
- Évaluez le petit modèle avec les données d'entraînement.
- Évaluez le petit modèle avec les données de test.
- Évaluez le grand modèle avec les données d'entraînement.
- Évaluez le grand modèle avec les données de test.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Evaluate the small model using the train data
small_train = small_model.evaluate(____, ____)
# Evaluate the small model using the test data
small_test = ____
# Evaluate the large model using the train data
large_train = large_model.evaluate(____, ____)
# Evaluate the large model using the test data
large_test = ____
# Print losses
print('\n Small - Train: {}, Test: {}'.format(small_train, small_test))
print('Large - Train: {}, Test: {}'.format(large_train, large_test))