otherModelForComparison.py 文件源码

python
阅读 22 收藏 0 点赞 0 评论 0

项目:CreditScoring 作者: cqw5 项目源码 文件源码
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
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号