shared.py

from pathlib import Path



import pandas as pd

splits = {'train': 'IMDB_train.csv', 'validation': 'IMDB_validation.csv', 'test': 'IMDB_test.csv'}
df = pd.read_csv("hf://datasets/jahjinx/IMDb_movie_reviews/" + splits["train"])