Refer to HW for the train and test dataset.

See the `sample_submission.csv` below for what an example submission looks like.