fixed notebook saving and loading code

Author: cpmpercussion

.gitignore

@@ -2,6 +2,7 @@
 *.svg
 *.npz
 *.h5
+*.keras
 
 # Byte-compiled / optimized / DLL files
 __pycache__/

notebooks/MDN-1D-sine-prediction.ipynb

@@ -113,7 +113,7 @@
    "outputs": [],
    "source": [
     "# Save model if you want to.\n",
-    "# model.save(\"MDN-1D-sine-prediction-model.h5\")"
+    "model.save(\"MDN-1D-sine-prediction-model.keras\")"
    ]
   },
   {
@@ -124,7 +124,7 @@
    "source": [
     "# Load the model if you want to.\n",
     "# To load models from file, you need to supply the layer and loss function as custom_objects:\n",
-    "# model = keras.models.load_model('MDN-1D-sine-prediction-model.h5', custom_objects={'MDN': mdn.MDN, 'loss_func': mdn.get_mixture_loss_func(1, N_MIXES)})"
+    "model = keras.models.load_model('MDN-1D-sine-prediction-model.keras', custom_objects={'MDN': mdn.MDN, 'mdn_loss_func': mdn.get_mixture_loss_func(1, N_MIXES)})"
    ]
   },
   {