def train_model(ngrams_file, output_file): model = Bayes() model.train(Ngrams._load_data(ngrams_file)) model.serialize(output_file)