Source code for epbd_bert.datasets.sequence_randepbd_dataset

import torch
import transformers

from epbd_bert.datasets.sequence_epbd_dataset import SequenceEPBDDataset


[docs] class SequenceRandEPBDDataset(SequenceEPBDDataset): def __init__( self, data_path: str, tokenizer: transformers.PreTrainedTokenizer, home_dir="" ): super().__init__(data_path, tokenizer, home_dir) def _get_epbd_features(self, fname): return torch.rand(1200, dtype=torch.float32)