def build_naive_bayes_model(x_train, y_train): nb_model = GaussianNB() nb_model.fit(x_train,y_train.ravel()) return nb_model