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 de ejercicio
- Configura el optimizador para que realice la minimización.
- Añade las cuatro variables entrenables a
var_list
en el orden en que aparecen como argumentos deloss_function().
- Utiliza el modelo y
test_features
para predecir los valores detest_targets
.
Ejercicio interactivo práctico
Pruebe este ejercicio completando este 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)