Fixes to the TensorFlow conversion tool

2019-04-01 12:53:51 -06:00
parent ec5c1d6134
commit 8b5c63e4de
2 changed files with 7 additions and 3 deletions
--- a/examples/extract_features.py
+++ b/examples/extract_features.py
@@ -57,7 +57,7 @@ class InputFeatures(object):


 def convert_examples_to_features(examples, seq_length, tokenizer):
-    """Loads a data file into a list of `InputBatch`s."""
+    """Loads a data file into a list of `InputFeature`s."""

    features = []
    for (ex_index, example) in enumerate(examples):