Analysis started, almost complete - compiled some excel sheets from the csv output with notes. Started infer.py, nothing major implemented yet
This commit is contained in:
@@ -51,7 +51,7 @@ 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))
|
||||
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user