Machine Learning机器学习 - Supervised Learning监督学习 - Native Bayes

最新推荐文章于 2024-10-02 10:53:34 发布

缘由的缘由

最新推荐文章于 2024-10-02 10:53:34 发布

阅读量497

点赞数

分类专栏： Machine Learning 文章标签： python 机器学习

本文链接：https://blog.csdn.net/shuizongbei/article/details/50382158

版权

Machine Learning 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

理论篇 Bayes Theorem

在维基百科上Bayes Theorem的解释已经很全面，特别是Drug Testing那段挺精彩，是Bayes Theorem理解的核心。
参考文章数学之美番外篇：平凡而又神奇的贝叶斯方法【中文】，有更精彩的描述。
上篇文章中提到的这篇文章How to Write a Spelling Corrector也很有启发。

实现篇 Python - 调用scikit Native Bayes库函数

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()

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

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

    ### calculate and return the accuracy on the test data
    from sklearn.metrics import accuracy_score
    accuracy = accuracy_score(labels_test, pred)
    return accuracy