Next, we split the data into train and test datasets: