(self)
| 534 | print("Label features added!") |
| 535 | |
| 536 | def text_emb(self): |
| 537 | text_model = TextModel(self.args.text_encoder) |
| 538 | text_model = text_model.to(self.args.device) |
| 539 | raw_texts = self.data.raw_texts |
| 540 | text_features = [] |
| 541 | for text in tqdm.tqdm(raw_texts, desc="Processing texts"): |
| 542 | text_features.append(text_model(text).unsqueeze(dim=0).cpu()) |
| 543 | self.data.x = torch.cat(text_features, dim=0) |
| 544 | print("Text features added!") |
| 545 | |
| 546 | @property |
| 547 | def raw_dir(self): |