Updated NaiveBayes training methods