(self, stage)
| 142 | |
| 143 | # encode the sentences using the tokenizer |
| 144 | def setup(self, stage): |
| 145 | self.train = self.encode_sentences(self.train['positives'], self.train['negatives']) |
| 146 | |
| 147 | self.validate = self.encode_sentences(self.validate['positives'], self.validate['negatives']) |
| 148 | |
| 149 | self.test = self.encode_sentences(self.test['positives'], self.test['negatives']) |
| 150 | |
| 151 | # Load the training, validation and test sets in Pytorch Dataset objects |
| 152 | def train_dataloader(self): |
nothing calls this directly
no test coverage detected