(self, idx)
| 63 | return len(self.dataset) |
| 64 | |
| 65 | def __getitem__(self, idx): |
| 66 | item = self.dataset[idx][0] |
| 67 | return (item[:self.seq_len],) if self.seq_len <= len(item) else ( |
| 68 | np.concatenate((item, np.ones(self.seq_len - len(item)) * self.eod_id), axis=0),) |
| 69 | # return (np.pad(item, (0, 1), constant_values=self.eod_id),) |
| 70 | |
| 71 | |
| 72 | class BinaryDataset(Dataset): |
nothing calls this directly
no outgoing calls
no test coverage detected