otherModelForComparison.py 文件源码

def sub_NB(train_x, train_y, test_x, test_y):
    """ ???????? """
    classifier = GaussianNB()
    classifier.fit(train_x, train_y)
    pred = classifier.predict_proba(test_x)
    predict_pro = []
    for pro in pred:
        predict_pro.append(pro[1])
    return np.array(predict_pro)