calculate and return the accuracy on the test data

def NBAccuracy(features_train, labels_train, features_test, labels_test):
    """ compute the accuracy of your Naive Bayes classifier """
    ### import the sklearn module for GaussianNB
    from sklearn.naive_bayes import GaussianNB

    ### create classifier
    clf = GaussianNB()  # TODO

    ### fit the classifier on the training features and labels
    clf.fit(features_train, labels_train)  # TODO

    ### use the trained classifier to predict labels for the test features
    pred = clf.predict(features_test)  # TODO
    print pred

    ### calculate and return the accuracy on the test data
    ### this is slightly different than the example,
    ### where we just print the accuracy
    ### you might need to import an sklearn module
#method 1
    accuracy = clf.score(features_test, labels_test)  # TODO
    print accuracy

#method 2
    from sklearn.metrics import accuracy_score
    accuracy = accuracy_score(pred, labels_test)
    print accuracy

    return accuracy
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值