def test_update_model(): bayes = test_NaiveBayes() print bayes.model arr = get_nb_data() print arr truth_rows, data_rows, data_mus, y_mu = hw3_run.get_data_and_mus(arr) bayes.train(data_rows, truth_rows) print bayes.model
def test_get_data_and_mus(): arr = get_test_data(4) truth_rows, data_rows, data_mus, y_mu = hw3_run.get_data_and_mus(arr) print 'unsplit {}'.format(arr) print 'truth rows {}'.format(truth_rows) print 'data rows {}'.format(data_rows) print 'data mus {}'.format(data_mus) print 'y_mu {}'.format(y_mu)
def test_update_model_change(): bayes = test_NaiveBayes() print bayes.model arr = get_nb_data(5) print arr truth_rows, data_rows, data_mus, y_mu = hw3_run.get_data_and_mus(arr) bayes.train(data_rows, truth_rows) print bayes.model test = get_nb_test_data(5) print 'prediction' print bayes.predict(test)