多项式朴素贝叶斯#

class sklearn.naive_bayes.MultinomialNB(*, alpha=1.0, force_alpha=True, fit_prior=True, class_prior=None)[source]#

用于多项式模型的朴素贝叶斯分类器。

多项式朴素贝叶斯分类器适用于具有离散特征的分类（例如，文本分类的词频）。多项式分布通常需要整数特征计数。但是，实际上，分数计数（例如 tf-idf）也可以工作。

更多信息请参见用户指南。

参数:
alpha浮点数或形状为 (n_features,) 的数组，默认为 1.0

附加 (拉普拉斯/利德斯通) 平滑参数（设置为 alpha=0 且 force_alpha=True，则不进行平滑）。

force_alpha布尔值，默认为 True

如果为 False 且 alpha 小于 1e-10，则将其设置为 1e-10。如果为 True，则 alpha 保持不变。如果 alpha 太接近 0，这可能会导致数值错误。

1.2 版本中新增。

1.4 版本中的更改： force_alpha 的默认值更改为 True。

fit_prior布尔值，默认为 True

是否学习类先验概率。如果为 False，则使用均匀先验。

class_prior形状为 (n_classes,) 的数组，默认为 None

类的先验概率。如果指定，则不会根据数据调整先验。

属性:
class_count_形状为 (n_classes,) 的 ndarray

拟合过程中遇到的每个类的样本数。提供样本权重时，此值将按样本权重加权。

class_log_prior_形状为 (n_classes,) 的 ndarray

每个类的平滑经验对数概率。

classes_形状为 (n_classes,) 的 ndarray

分类器已知的类标签

feature_count_形状为 (n_classes, n_features) 的 ndarray

拟合过程中遇到的每个 (类，特征) 的样本数。提供样本权重时，此值将按样本权重加权。

feature_log_prob_形状为 (n_classes, n_features) 的 ndarray

给定类的特征的经验对数概率，P(x_i|y)。

n_features_in_整数

在拟合期间看到的特征数量。

0.24 版本中新增。

feature_names_in_形状为 (n_features_in_,) 的 ndarray

在拟合期间看到的特征名称。仅当 X 具有全是字符串的特征名称时定义。

1.0 版本中新增。

另请参见

伯努利朴素贝叶斯

用于多元伯努利模型的朴素贝叶斯分类器。

类别朴素贝叶斯

用于分类特征的朴素贝叶斯分类器。

补集朴素贝叶斯

补充朴素贝叶斯分类器。

高斯朴素贝叶斯

高斯朴素贝叶斯。

参考文献

C.D. Manning, P. Raghavan 和 H. Schuetze (2008)。信息检索导论。剑桥大学出版社，第 234-265 页。https://nlp.stanford.edu/IR-book/html/htmledition/naive-bayes-text-classification-1.html

示例

>>> import numpy as np
>>> rng = np.random.RandomState(1)
>>> X = rng.randint(5, size=(6, 100))
>>> y = np.array([1, 2, 3, 4, 5, 6])
>>> from sklearn.naive_bayes import MultinomialNB
>>> clf = MultinomialNB()
>>> clf.fit(X, y)
MultinomialNB()
>>> print(clf.predict(X[2:3]))
[3]
fit(X, y, sample_weight=None)[source]#

根据 X、y 拟合朴素贝叶斯分类器。

参数:
X形状为 (n_samples, n_features) 的{数组、稀疏矩阵}

训练向量，其中 n_samples 是样本数，n_features 是特征数。

y形状为 (n_samples,) 的数组

目标值。

sample_weight形状为 (n_samples,) 的数组，默认为 None

应用于各个样本的权重（对于未加权的样本为 1）。

返回值:
self对象

返回实例本身。

get_metadata_routing()[source]#

获取此对象的元数据路由。

请查看用户指南，了解路由机制的工作原理。

返回值:
routingMetadataRequest

一个MetadataRequest，封装了路由信息。

get_params(deep=True)[source]#

获取此估计器的参数。

参数:
deep布尔值，默认为 True

如果为 True，则将返回此估计器和包含的作为估计器的子对象的参数。

返回值:
params字典

参数名称与其值的映射。

partial_fit(X, y, classes=None, sample_weight=None)[source]#

对一批样本进行增量拟合。

预期此方法将连续多次在数据集的不同块上调用，以便实现核心外学习或在线学习。

当整个数据集太大而无法一次全部放入内存时，这尤其有用。

此方法有一些性能开销，因此最好对尽可能大的数据块（只要能放入内存预算）调用 partial_fit 来隐藏开销。

参数:
X形状为 (n_samples, n_features) 的{数组、稀疏矩阵}

训练向量，其中 n_samples 是样本数，n_features 是特征数。

y形状为 (n_samples,) 的数组

目标值。

classes形状为 (n_classes,) 的数组，默认为 None

y 向量中可能出现的所有类的列表。

必须在第一次调用 partial_fit 时提供，后续调用可以省略。

sample_weight形状为 (n_samples,) 的数组，默认为 None

应用于各个样本的权重（对于未加权的样本为 1）。

返回值:
self对象

返回实例本身。

predict(X)[source]#

对测试向量 X 数组执行分类。

参数:
X形状为 (n_samples, n_features) 的数组

输入样本。

返回值:
C形状为 (n_samples,) 的 ndarray

X 的预测目标值。

predict_joint_log_proba(X)[source]#

返回测试向量 X 的联合对数概率估计。

对于 X 的每一行 x 和类别 y，联合对数概率由 log P(x, y) = log P(y) + log P(x|y),给出，其中 log P(y) 是类别先验概率，log P(x|y) 是类条件概率。

参数:
X形状为 (n_samples, n_features) 的数组

输入样本。

返回值:
C形状为 (n_samples, n_classes) 的 ndarray

返回模型中每个类别的样本联合对数概率。各列对应于已排序的类别，其顺序与属性 classes_ 中的顺序相同。

predict_log_proba(X)[source]#

返回测试向量 X 的对数概率估计。

参数:
X形状为 (n_samples, n_features) 的数组

输入样本。

返回值:
C形状为 (n_samples, n_classes) 的数组型对象

返回模型中每个类别的样本对数概率。各列对应于已排序的类别，其顺序与属性 classes_ 中的顺序相同。

predict_proba(X)[source]#

返回测试向量 X 的概率估计。

参数:
X形状为 (n_samples, n_features) 的数组

输入样本。

返回值:
C形状为 (n_samples, n_classes) 的数组型对象

返回模型中每个类别的样本概率。各列对应于已排序的类别，其顺序与属性 classes_ 中的顺序相同。

score(X, y, sample_weight=None)[source]#

返回给定测试数据和标签的平均准确率。

在多标签分类中，这是子集准确率，这是一个严格的指标，因为您要求每个样本的每个标签集都必须被正确预测。

参数:
X形状为 (n_samples, n_features) 的数组

测试样本。

y形状为 (n_samples,) 或 (n_samples, n_outputs) 的数组型对象

X 的真实标签。

sample_weight形状为 (n_samples,) 的数组，默认为 None

样本权重。

返回值:
score浮点数

self.predict(X) 关于 y 的平均准确率。

set_fit_request(*, sample_weight: bool | None | str = '$UNCHANGED$') → MultinomialNB[source]#

请求传递到 fit 方法的元数据。

请注意，只有在 enable_metadata_routing=True 时，此方法才相关（请参见 sklearn.set_config）。请参阅 用户指南，了解路由机制的工作原理。

每个参数的选项为

  • True：请求元数据，如果提供则传递到 fit。如果未提供元数据，则忽略请求。

  • False：不请求元数据，元估计器不会将其传递给 fit。

  • None：不请求元数据，如果用户提供元数据，元估计器将引发错误。

  • str：元数据应使用此给定的别名而不是原始名称传递给元估计器。

默认值（sklearn.utils.metadata_routing.UNCHANGED）保留现有的请求。这允许您更改某些参数的请求，而无需更改其他参数的请求。

版本 1.3 中新增。

注意

仅当将此估计器用作元估计器的子估计器（例如，在 Pipeline 中使用）时，此方法才相关。否则，它没有任何作用。

参数:
sample_weightstr、True、False 或 None，默认为 sklearn.utils.metadata_routing.UNCHANGED

fit 中 sample_weight 参数的元数据路由。

返回值:
self对象

更新后的对象。

set_params(**params)[source]#

设置此估计器的参数。

此方法适用于简单的估计器以及嵌套对象（例如 Pipeline）。后者具有 <component>__<parameter> 形式的参数，因此可以更新嵌套对象的每个组件。

参数:
**params**dict

估计器参数。

返回值:
selfestimator instance

估计器实例。

set_partial_fit_request(*, classes: bool | None | str = '$UNCHANGED$', sample_weight: bool | None | str = '$UNCHANGED$') → MultinomialNB[source]#

请求传递给 partial_fit 方法的元数据。

请注意，只有在 enable_metadata_routing=True 时，此方法才相关（请参见 sklearn.set_config）。请参阅 用户指南，了解路由机制的工作原理。

每个参数的选项为

  • True：请求元数据，如果提供则传递给 partial_fit。如果未提供元数据，则忽略请求。

  • False：不请求元数据，元估计器不会将其传递给 partial_fit。

  • None：不请求元数据，如果用户提供元数据，元估计器将引发错误。

  • str：元数据应使用此给定的别名而不是原始名称传递给元估计器。

默认值（sklearn.utils.metadata_routing.UNCHANGED）保留现有的请求。这允许您更改某些参数的请求，而无需更改其他参数的请求。

版本 1.3 中新增。

注意

仅当将此估计器用作元估计器的子估计器（例如，在 Pipeline 中使用）时，此方法才相关。否则，它没有任何作用。

参数:
classesstr, True, False 或 None，默认为 sklearn.utils.metadata_routing.UNCHANGED

partial_fit 方法中 classes 参数的元数据路由。

sample_weightstr、True、False 或 None，默认为 sklearn.utils.metadata_routing.UNCHANGED

partial_fit 方法中 sample_weight 参数的元数据路由。

返回值:
self对象

更新后的对象。

set_score_request(*, sample_weight: bool | None | str = '$UNCHANGED$') → MultinomialNB[source]#

请求传递给 score 方法的元数据。

请注意，只有在 enable_metadata_routing=True 时，此方法才相关（请参见 sklearn.set_config）。请参阅 用户指南，了解路由机制的工作原理。

每个参数的选项为

  • True：请求元数据，如果提供则传递给 score。如果未提供元数据，则忽略请求。

  • False：不请求元数据，元估计器不会将其传递给 score。

  • None：不请求元数据，如果用户提供元数据，元估计器将引发错误。

  • str：元数据应使用此给定的别名而不是原始名称传递给元估计器。

默认值（sklearn.utils.metadata_routing.UNCHANGED）保留现有的请求。这允许您更改某些参数的请求，而无需更改其他参数的请求。

版本 1.3 中新增。

注意

仅当将此估计器用作元估计器的子估计器（例如，在 Pipeline 中使用）时，此方法才相关。否则，它没有任何作用。

参数:
sample_weightstr、True、False 或 None，默认为 sklearn.utils.metadata_routing.UNCHANGED

score 方法中 sample_weight 参数的元数据路由。

返回值:
self对象

更新后的对象。