Don't always split the train set 80-20, allow the caller to specify a train and validation set.
Don't always split the train set 80-20, allow the caller to specify a train and validation set.