Fixing errors in docs

theainerd · Jun 9, 2020 · d3320cc · d3320cc
1 parent f535090
commit d3320cc
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/docs/source/data_transformations.rst b/docs/source/data_transformations.rst
@@ -15,7 +15,7 @@ Sample transform functions
 ^^^^^^^^^^^^^^^^^^^^^^^^^^
 .. automodule:: utils.tranform_functions
     :members: snips_intent_ner_to_tsv, bio_ner_to_tsv, snli_entailment_to_tsv,
-        fragment_detection_to_tsv, msmarco_query_type_to_tsv
+        create_fragment_detection_tsv, msmarco_query_type_to_tsv, qqp_query_similarity_to_tsv
 
 Your own transform function
 ^^^^^^^^^^^^^^^^^^^^^^^^^^^

diff --git a/docs/source/shared_encoder.rst b/docs/source/shared_encoder.rst
@@ -78,5 +78,6 @@ Metrics
 For evaluating the performance on dev and test sets during training, we provide the following standard metrics.
 
 .. automodule:: utils.eval_metrics
-    :members:
+    :members: classification_accuracy, classification_f1_score, seqeval_f1_score, 
+        seqeval_precision, seqeval_recall, snips_f1_score, snips_precision, snips_recall
 
diff --git a/utils/eval_metrics.py b/utils/eval_metrics.py
@@ -1,6 +1,4 @@
-"""
-File for creating metric functions
-"""
+
 from sklearn.metrics import accuracy_score, f1_score
 from seqeval.metrics import f1_score as seq_f1
 from seqeval.metrics import precision_score, recall_score