Fixed generate_classification_plots

Palashio · Sep 22, 2020 · 199ddde · 199ddde
1 parent edeb8e8
commit 199ddde
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/libra/query/nlp_queries.py b/libra/query/nlp_queries.py
@@ -208,8 +208,7 @@ def text_classification_query(self, instruction, drop=None,
         # generates appropriate classification plots by feeding all
         # information
         logger("Generating plots")
-        plots = generate_classification_plots(
-            history, X, Y, model, X_test, y_test)
+        plots = generate_classification_plots(history)
 
     if save_model:
         save(model, save_model, save_path=save_path)

diff --git a/libra/query/supplementaries.py b/libra/query/supplementaries.py
@@ -177,8 +177,7 @@ def tune_helper(
         logger("->", 'Best Hyperparameters Found: {}'.format(returned_pms.values))
         if generate_plots:
             logger("Generating updated plots")
-            plots = generate_classification_plots(
-                    history, data, target_column, returned_model, X_test, y_test)
+            plots = generate_classification_plots(history)
 
 
         logger("Re-stored model under 'classification_ANN' key")