Added training loop for the MTL architecture on the original distribution

2026-02-23 16:26:48 +00:00
parent 7bd68108d0
commit 4f0c54fe28
8 changed files with 174 additions and 37 deletions
--- a/src/dataset.py
+++ b/src/dataset.py
@@ -51,7 +51,8 @@ class ReviewDataset(Dataset):
    
 if __name__ == "__main__":
    dataset = ReviewDataset("data/processed/original_train.csv", AutoTokenizer.from_pretrained("FacebookAI/xlm-roberta-base"))
-    print(dataset.__getitem__(1))
+    # print(dataset.__getitem__(1))
+