Entraîner des réseaux de neurones avec TensorFlow
Dans l'exercice précédent, vous avez défini un modèle, model(w1, b1, w2, b2, features), et une fonction de perte, loss_function(w1, b1, w2, b2, features, targets), tous deux à votre disposition ici. Vous allez maintenant entraîner le modèle, puis évaluer sa performance en prédisant les défauts dans un jeu de test, composé de test_features et test_targets, également à votre disposition. Les variables entraînables sont w1, b1, w2 et b2. De plus, les opérations suivantes ont été importées pour vous : keras.activations.relu() et keras.layers.Dropout().
Cette activité fait partie du cours
Introduction à TensorFlow en Python
Instructions de l’exercice
- Paramétrez l'optimiseur pour effectuer la minimisation.
- Ajoutez les quatre variables entraînables à
var_listdans l'ordre où elles apparaissent comme arguments deloss_function(). - Utilisez le modèle et
test_featurespour prédire les valeurs detest_targets.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Train the model
for j in range(100):
# Complete the optimizer
opt.____(lambda: loss_function(w1, b1, w2, b2),
var_list=[____, ____, ____, ____])
# Make predictions with model using test features
model_predictions = model(w1, b1, w2, b2, ____)
# Construct the confusion matrix
confusion_matrix(test_targets, model_predictions)