Compare commits

..

1 Commits

Author SHA1 Message Date
coolneng 07776a0538
Perform one hot encoding on the sequences 2021-06-25 00:04:01 +02:00
1 changed files with 1 additions and 1 deletions

View File

@ -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 one hot encoded Tensors
Transform the parsed features of an Example into a list of dense Tensors
"""
features = {}
sparse_features = ["sequence", "label"]