Add hold out mechanism
Gaussian Naive Bayes for training data