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_prior：array-like of shape (n_classes,), default=None: 类的先验概率。如果指定，则不会根据数据调整先验。
output_type：{‘input’, ‘cudf’, ‘cupy’, ‘numpy’, ‘numba’}，默认=无: 用于控制估计器的结果和属性的输出类型的变量。如果为 None，它将继承在模块级别设置的输出类型 cuml.global_settings.output_type 。有关详细信息，请参阅输出数据类型配置。
handle：cuml.Handle: 指定 cuml.handle 保存用于此模型中计算的内部 CUDA 状态。最重要的是，这指定了将用于模型计算的 CUDA 流，因此用户可以通过在多个流中创建句柄在不同的流中同时运行不同的模型。如果为 None，则创建一个新的。
verbose：int 或布尔值，默认=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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。