Move get_f1_score() to train.py
load_data() can return the dataset for just one class now
Tidied up
Using enum instead of strings to identify temporal relations