[no ci] docstring and warning edits

amaiya · Jun 14, 2023 · 88d2bb5 · 88d2bb5
1 parent 64ed862
commit 88d2bb5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/ktrain/core.py b/ktrain/core.py
@@ -134,8 +134,8 @@ def validate(
         ```
         Returns confusion matrix and optionally prints
         a classification report.
-        This is currently only supported for binary and multiclass
-        classification, not multilabel classification.
+        For multilabel classification problems,confusion matrices are not supported,
+        but classification reports are.
 
         By default, this uses val_data, as supplied to ktrain.get_learner().
         Other validation or test data can be optionally be supplied as argument.

diff --git a/ktrain/text/preprocessor.py b/ktrain/text/preprocessor.py
@@ -1141,7 +1141,8 @@ def get_classifier(self, fpath=None, multilabel=None, metrics=["accuracy"]):
 
         if multilabel and metrics == ["accuracy"]:
             warnings.warn(
-                'For multilabel problems, we recommend you supply the following argument to this method: metrics=["binary_accuracy"]'
+                'For multilabel problems, we recommend you supply the following argument to this method: metrics=["binary_accuracy"]. '
+                + "Otherwise, a low accuracy score will be displayed by TensorFlow (https://github.com/tensorflow/tensorflow/issues/41114)."
             )
 
         # setup model