Removed non variational autoencoder

main.ipynb

@@ -590,7 +590,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 147,
+   "execution_count": 152,
    "metadata": {},
    "outputs": [
     {
@@ -599,15 +599,15 @@
      "text": [
       "(690036, 11, 12)\n",
       "(14991, 11, 12)\n",
-      "CPU times: user 13.6 s, sys: 341 ms, total: 13.9 s\n",
-      "Wall time: 13.9 s\n"
+      "CPU times: user 16.5 s, sys: 396 ms, total: 16.9 s\n",
+      "Wall time: 17.7 s\n"
      ]
     }
    ],
    "source": [
     "%%time\n",
     "pttest, pttrain = mtest.preprocess(window_size=11, window_step=1, filter=None, sigma=55,\n",
-    "                          shift=0, scale=None, align='all', shuffle=True, test_videos=1)\n",
+    "                          shift=0, scale='standard', align='all', shuffle=True, test_videos=1)\n",
     "print(pttest.shape)\n",
     "print(pttrain.shape)"
    ]
@@ -645,7 +645,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 150,
+   "execution_count": 151,
    "metadata": {},
    "outputs": [
     {

source/model_utils.py

@@ -2,7 +2,6 @@
 
 from itertools import combinations
 from tensorflow.keras import backend as K
-from sklearn.metrics import silhouette_score
 from tensorflow.keras.constraints import Constraint
 from tensorflow.keras.layers import Layer
 import tensorflow as tf

source/models.py

@@ -380,7 +380,7 @@ class SEQ_2_SEQ_GMVAE:
             )(z)
 
         # Identity layer controlling clustering and latent space statistics
-        z = Dead_neuron_control(loss=self.overlap_loss)(z, z_gauss, z_cat)
+        z = Dead_neuron_control()(z, z_gauss, z_cat)
 
         # Define and instantiate generator
         generator = Model_D1(z)