Perform one hot encoding on the sequences

2021-06-25 00:05:14 +02:00
1 changed files with 1 additions and 1 deletions
--- a/src/preprocessing.py
+++ b/src/preprocessing.py
@ -77,7 +77,7 @@ def create_dataset(data_file, label_file, dataset_split=[0.8, 0.1, 0.1]) -> None
 def transform_features(parsed_features) -> Dict[str, Tensor]:
    """
-    Transform the parsed features of an Example into a list of dense Tensors
+    Transform the parsed features of an Example into a list of dense one hot encoded Tensors
    """
    features = {}
    sparse_features = ["sequence", "label"]