fix failed unittest

Author: williamgilpin

compare.ipynb

@@ -20,14 +20,26 @@
    "cell_type": "code",
    "execution_count": 1,
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stderr",
+     "output_type": "stream",
+     "text": [
+      "/Users/william/miniconda3/envs/nn2/lib/python3.6/site-packages/tensorflow_addons/utils/ensure_tf_install.py:44: UserWarning: You are currently using a nightly version of TensorFlow (2.4.0-dev20200804). \n",
+      "TensorFlow Addons offers no support for the nightly versions of TensorFlow. Some things might work, some other might not. \n",
+      "If you encounter a bug, do not file an issue on GitHub.\n",
+      "  UserWarning,\n"
+     ]
+    }
+   ],
    "source": [
     "import numpy as np\n",
     "\n",
     "## Import from local directory\n",
     "import sys\n",
     "sys.path.insert(0, './fnn')\n",
-    "from models import LSTMEmbedding, MLPEmbedding, ETDEmbedding, AMIEmbedding, TICAEmbedding, FNN\n",
+    "from models import LSTMEmbedding, MLPEmbedding, ETDEmbedding, AMIEmbedding, TICAEmbedding\n",
+    "from regularizers import FNN\n",
     "\n",
     "import matplotlib.pyplot as plt\n",
     "from mpl_toolkits.mplot3d import axes3d\n",
@@ -55,7 +67,7 @@
     {
      "data": {
       "text/plain": [
-       "[<matplotlib.lines.Line2D at 0x1538c7f28>]"
+       "[<matplotlib.lines.Line2D at 0x150ada748>]"
       ]
      },
      "execution_count": 2,

demos.ipynb

@@ -30,7 +30,8 @@
     "## Import from local directory\n",
     "import sys\n",
     "sys.path.insert(0, './fnn')\n",
-    "from models import LSTMEmbedding, MLPEmbedding, FNN\n",
+    "from models import LSTMEmbedding, MLPEmbedding\n",
+    "from regularizers import FNN\n",
     "\n",
     "plt.rcParams['lines.linewidth'] = .1\n",
     "plt.rcParams['axes.prop_cycle'] = plt.cycler(color=\"k\") \n",
@@ -62,7 +63,7 @@
     {
      "data": {
       "text/plain": [
-       "[<matplotlib.lines.Line2D at 0x15cbdbb00>]"
+       "[<matplotlib.lines.Line2D at 0x15ce04b70>]"
       ]
      },
      "execution_count": 2,

exploratory.ipynb

@@ -28,7 +28,8 @@
     "## Import from local directory\n",
     "import sys\n",
     "sys.path.insert(0, './fnn')\n",
-    "from models import LSTMEmbedding, MLPEmbedding, FNN\n",
+    "from models import LSTMEmbedding, MLPEmbedding\n",
+    "from regularizers import FNN\n",
     "\n",
     "import matplotlib.pyplot as plt\n",
     "from mpl_toolkits.mplot3d import axes3d\n",