def 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])
predict_y = classifier.predict(test_x)
auc = evaluate_auc(predict_pro, test_y)
evaluate(predict_y, test_y)
return auc
评论列表
文章目录