Use train/test split provided by the dataset #13

Closed hominot closed 5 years ago