agatha.ml.abstract_generator.datasets module¶
-
agatha.ml.abstract_generator.datasets.collate_encoded_abstracts(batch, key_subset=None)¶ - Return type
Dict[str,LongTensor]
-
agatha.ml.abstract_generator.datasets.shift_text_features_for_training(batch)¶ - Return type
Tuple[Dict[str,LongTensor],Dict[str,LongTensor]]