當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python cuml.naive_bayes.BernoulliNB用法及代碼示例


用法:

class cuml.naive_bayes.BernoulliNB(*, alpha=1.0, binarize=0.0, fit_prior=True, class_prior=None, output_type=None, handle=None, verbose=False)

用於多元伯努利模型的樸素貝葉斯分類器。與MultinomialNB 一樣,此分類器適用於離散數據。不同之處在於,雖然 MultinomialNB 適用於出現計數,但 BernoulliNB 是為二進製/布爾特征設計的。

參數

alpha浮點數,默認=1.0

Additive (Laplace/Lidstone) 平滑參數(0 表示無平滑)。

binarize浮點數或無,默認=0.0

樣本特征二值化(映射到布爾值)的閾值。如果沒有,則假定輸入已經由二進製向量組成。

fit_prior布爾,默認=真

是否學習類先驗概率。如果為 false,將使用統一的先驗。

class_priorarray-like of shape (n_classes,), default=None

類的先驗概率。如果指定,則不會根據數據調整先驗。

output_type{‘input’, ‘cudf’, ‘cupy’, ‘numpy’, ‘numba’},默認=無

用於控製估計器的結果和屬性的輸出類型的變量。如果為 None,它將繼承在模塊級別設置的輸出類型 cuml.global_settings.output_type 。有關詳細信息,請參閱輸出數據類型配置。

handlecuml.Handle

指定 cuml.handle 保存用於此模型中計算的內部 CUDA 狀態。最重要的是,這指定了將用於模型計算的 CUDA 流,因此用戶可以通過在多個流中創建句柄在不同的流中同時運行不同的模型。如果為 None,則創建一個新的。

verboseint 或布爾值,默認=False

設置日誌記錄級別。它必須是 cuml.common.logger.level_* 之一。有關詳細信息,請參閱詳細級別。

參考

光盤。 Manning、P. Raghavan 和 H. Schuetze(2008 年)。信息檢索導論。劍橋大學出版社,第 234-265 頁。https://nlp.stanford.edu/IR-book/html/htmledition/the-bernoulli-model-1.htmlA. McCallum 和 K. Nigam (1998)。樸素貝葉斯文本分類的事件模型比較。過程。 AAAI/ICML-98 文本分類學習研討會,第 41-48 頁。 V. Metsis、I. Androutsopoulos 和 G. Paliouras (2006)。使用樸素貝葉斯過濾垃圾郵件 - 哪種樸素貝葉斯?第三次會議。關於電子郵件和反垃圾郵件 (CEAS)。

例子

>>> import cupy as cp
>>> rng = cp.random.RandomState(1)
>>> X = rng.randint(5, size=(6, 100), dtype=cp.int32)
>>> Y = cp.array([1, 2, 3, 4, 4, 5])
>>> from cuml.naive_bayes import BernoulliNB
>>> clf = BernoulliNB()
>>> clf.fit(X, Y)
BernoulliNB()
>>> print(clf.predict(X[2:3]))
[3]

屬性

class_count_形狀的ndarray(n_classes)

擬合期間每個類遇到的樣本數。

class_log_prior_形狀的ndarray(n_classes)

每個類的對數概率(平滑)。

classes_ndarray 形狀 (n_classes,)

分類器已知的類標簽

feature_count_ndarray 形狀(n_classes,n_features)

擬合期間每個(類、特征)遇到的樣本數。

feature_log_prob_ndarray 形狀(n_classes,n_features)

給定類 P(x_i|y) 的特征的經驗對數概率。

n_features_int

每個樣本的特征數。

相關用法


注:本文由純淨天空篩選整理自rapids.ai大神的英文原創作品 cuml.naive_bayes.BernoulliNB。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。