Use 80% of the data set for training, 20% for testing