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)
评论列表
文章目录