Minor changes

tulane-cmps6730 · Apr 29, 2024 · 83df56e · 83df56e
1 parent 8060349
commit 83df56e
Show file tree

Hide file tree

Showing 4 changed files with 46 additions and 41 deletions.
diff --git a/nlp/cli.py b/nlp/cli.py
@@ -190,7 +190,11 @@ def train_cnn():
 
     pickle.dump(model, open(cnn_path, 'wb'))
 
-
+@main.command('train_cnn')
+def train_bert():
+    '''
+    Get BERT
+    '''
 
 
 if __name__ == "__main__":

diff --git a/notebooks/Experiment-CNN-1.ipynb b/notebooks/Experiment-CNN-1.ipynb
diff --git a/notebooks/Experiments-BERT.ipynb b/notebooks/Experiments-BERT.ipynb
@@ -119,14 +119,6 @@
     ")).batch(30).prefetch(1)\n"
    ]
   },
-  {
-   "cell_type": "markdown",
-   "id": "bdbd73a3-094c-4a8a-a45b-8ce8aecdcd12",
-   "metadata": {},
-   "source": [
-    "### Without Custom Classifier"
-   ]
-  },
   {
    "cell_type": "code",
    "execution_count": 15,

diff --git a/notebooks/accuracy_plot.png b/notebooks/accuracy_plot.png