benchmarks/imdb/scripts/research_problem.txt

Fill out train.py to 1) finetune DistilBERT on the IMDb dataset to determine whether a movie review is positive or negative and 2) save per class probabilities for test set examples to submission.csv. The dataset is available via HuggingFace's datasets library. You can find skeleton code for loading the dataset and saving prediction to subission.csv in `train.py`.