Started dataset.py, added the ReviewDataset class and implemented the __init__, __len__ and __getitem__ methods. The __getitem__ method currently just returns the review text, but will be updated to return the tokenized review as a tensor

2026-02-19 18:45:55 +00:00
parent 19bcf2aa18
commit 19c0d4bce3
2 changed files with 2 additions and 1 deletions
--- a/src/dataset.py
+++ b/src/dataset.py
@@ -1,7 +1,6 @@
 # dataset.py

 # Takes a row from the csv, tokenizes the review and returns a tensor
-
 import torch
 import pandas as pd
 from torch.utils.data import Dataset