Add docstring for processor method

This commit is contained in:
Agrin Hilmkil
2019-09-27 17:32:28 +02:00
parent e31a472801
commit 795b3e76ff
2 changed files with 18 additions and 0 deletions

View File

@@ -86,6 +86,15 @@ class InputFeatures(object):
class DataProcessor(object):
"""Base class for data converters for sequence classification data sets."""
def get_example_from_tensor_dict(self, tensor_dict):
"""Gets an example from a dict with tensorflow tensors
Args:
tensor_dict: Keys and values should match the corresponding Glue
tensorflow_dataset examples.
"""
raise NotImplementedError()
def get_train_examples(self, data_dir):
"""Gets a collection of `InputExample`s for the train set."""
raise NotImplementedError()