Rename

tulane-cmps6730 · Apr 28, 2024 · 07b0e9a · 07b0e9a
1 parent 45c0f97
commit 07b0e9a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/nlp/cli.py b/nlp/cli.py
@@ -68,8 +68,8 @@ def data2df():
 
 
 
-@main.command('train_bn')
-def train_nb(train_df, val_df):
+@main.command('train_nb')
+def train_nb():
     """
     Naive Bayes Model (Do Third)
     """
@@ -107,7 +107,7 @@ def process_text(document):
     print("F1 Score:", round(f1,3))
 
 @main.command('train_lr')
-def train_lr(train_df, val_df):
+def train_lr():
     """
     Logistic Regression Model (Do Fourth)
     """