create empty list

Author: shincap8

supervised_learning/0x06-keras/6-train.py

@@ -1,5 +1,6 @@
 #!/usr/bin/env python3
-"""Function that trains a model using mini-batch gradient descent"""
+"""Function that trains a model using mini-batch gradient descent
+also analyze validaiton data and early stopping"""
 
 import tensorflow.keras as K
 
@@ -8,11 +9,10 @@ def train_model(network, data, labels, batch_size, epochs,
                 validation_data=None, early_stopping=False,
                 patience=0, verbose=True, shuffle=False):
     """return history"""
+    callback = []
     if validation_data is not None and early_stopping:
-        callback = [K.callbacks.EarlyStopping(monitor="val_loss",
-                                              patience=patience)]
-    else:
-        callback = NULL
+        callback.append(K.callbacks.EarlyStopping(monitor="val_loss",
+                                                  patience=patience))
     history = network.fit(
         x=data,
         y=labels,