Entrenar redes neuronales con TensorFlow
En el ejercicio anterior, definiste un modelo, model(w1, b1, w2, b2, features), y una función de pérdida, loss_function(w1, b1, w2, b2, features, targets), ambos disponibles en este ejercicio. Ahora entrenarás el modelo y luego evaluarás su rendimiento prediciendo resultados predeterminados en un conjunto de pruebas, que consta de test_features y test_targets y está a tu disposición. Las variables entrenables son w1, b1, w2, y b2. Además, se han importado para ti las siguientes operaciones: keras.activations.relu() y keras.layers.Dropout().
Este ejercicio forma parte del curso
Introducción a TensorFlow en Python
Instrucciones del ejercicio
- Configura el optimizador para que realice la minimización.
- Añade las cuatro variables entrenables a
var_listen el orden en que aparecen como argumentos deloss_function(). - Utiliza el modelo y
test_featurespara predecir los valores detest_targets.
Ejercicio interactivo práctico
Prueba este ejercicio y completa el código de muestra.
# 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)