DictVectorizer#

class sklearn.feature_extraction.DictVectorizer(*, dtype=<class 'numpy.float64'>, separator='=', sparse=True, sort=True)[source]#

将特征-值映射列表转换为向量。

此转换器将特征名称到特征值的映射列表(类似字典的对象)转换为NumPy数组或scipy.sparse矩阵,以便与scikit-learn估计器一起使用。

当特征值是字符串时,此转换器将执行二元独热编码(又称one-of-K编码):为特征可能采用的每个可能的字符串值构造一个布尔值特征。例如,一个可以取值“ham”和“spam”的特征“f”将变成输出中的两个特征,一个表示“f=ham”,另一个表示“f=spam”。

如果特征值是字符串序列或集合,此转换器将迭代这些值并计算每个字符串值的出现次数。

但是,请注意,只有当特征值的数据类型为字符串时,此转换器才会执行二元独热编码。如果分类特征表示为数值(例如整数)或字符串迭代器,则可以使用 OneHotEncoder 来完成二元独热编码。

样本(映射)中未出现的特征在生成的数组/矩阵中将具有零值。

有关不同特征提取器的效率比较,请参阅 FeatureHasher 和 DictVectorizer 比较

用户指南 中了解更多信息。

参数:
dtypedtype,默认为np.float64

特征值的类型。作为dtype参数传递给Numpy数组/scipy.sparse矩阵构造函数。

separatorstr,默认为“=”

构造独热编码的新特征时使用的分隔符字符串。

sparsebool,默认为True

转换是否应生成scipy.sparse矩阵。

sortbool,默认为True

拟合时是否应对feature_names_vocabulary_进行排序。

属性:
vocabulary_dict

一个将特征名称映射到特征索引的字典。

feature_names_list

一个长度为n_features的列表,包含特征名称(例如,“f=ham”和“f=spam”)。

另请参阅

FeatureHasher

仅使用哈希函数执行向量化。

sklearn.preprocessing.OrdinalEncoder

处理编码为任意数据类型列的标称/分类特征。

示例

>>> from sklearn.feature_extraction import DictVectorizer
>>> v = DictVectorizer(sparse=False)
>>> D = [{'foo': 1, 'bar': 2}, {'foo': 3, 'baz': 1}]
>>> X = v.fit_transform(D)
>>> X
array([[2., 0., 1.],
       [0., 1., 3.]])
>>> v.inverse_transform(X) == [{'bar': 2.0, 'foo': 1.0},
...                            {'baz': 1.0, 'foo': 3.0}]
True
>>> v.transform({'foo': 4, 'unseen_feature': 3})
array([[0., 0., 4.]])
fit(X, y=None)[source]#

学习特征名称到索引映射的列表。

参数:
X映射或可迭代的映射

从特征名称(任意Python对象)到特征值(字符串或可转换为dtype)的字典或映射。

0.24版本中的更改: 接受单个分类特征的多个字符串值。

y(忽略)

忽略的参数。

返回值:
self对象

DictVectorizer 类实例。

fit_transform(X, y=None)[source]#

学习特征名称到索引映射的列表并转换X。

类似于fit(X) 后跟 transform(X),但不需要在内存中具体化X。

参数:
X映射或可迭代的映射

从特征名称(任意Python对象)到特征值(字符串或可转换为dtype)的字典或映射。

0.24版本中的更改: 接受单个分类特征的多个字符串值。

y(忽略)

忽略的参数。

返回值:
Xa{数组,稀疏矩阵}

特征向量;始终为二维。

get_feature_names_out(input_features=None)[source]#

获取转换的输出特征名称。

参数:
input_featuresstr 的类数组或 None,默认为 None

未使用,此处出于 API 一致性约定而存在。

返回值:
feature_names_outstr 对象的 ndarray

转换后的特征名称。

get_metadata_routing()[source]#

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

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

返回值:
routingMetadataRequest

一个 MetadataRequest 封装路由信息。

get_params(deep=True)[source]#

获取此估计器的参数。

参数:
deepbool,默认为True

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

返回值:
paramsdict

参数名称映射到其值。

inverse_transform(X, dict_type=<class 'dict'>)[source]#

将数组或稀疏矩阵 X 转换回特征映射。

X 必须是由此 DictVectorizer 的 transform 或 fit_transform 方法生成的;它可能仅通过保留特征数量及其顺序的转换器。

对于独热编码/K值独热编码,返回的是构造的特征名称和值,而不是原始的名称和值。

参数:
X形状为 (n_samples, n_features) 的 {array-like, sparse matrix}

样本矩阵。

dict_type类型,默认为 dict

特征映射的构造器。必须符合 collections.Mapping API。

返回值:
D形状为 (n_samples,) 的 dict_type 对象列表

X 中样本的特征映射。

restrict(support, indices=False)[source]#

使用特征选择将特征限制在 support 中的那些特征。

此函数就地修改估计器。

参数:
support类数组

布尔掩码或索引列表(由特征选择器的 get_support 成员返回)。

indices布尔值,默认为 False

support 是否为索引列表。

返回值:
self对象

DictVectorizer 类实例。

示例

>>> from sklearn.feature_extraction import DictVectorizer
>>> from sklearn.feature_selection import SelectKBest, chi2
>>> v = DictVectorizer()
>>> D = [{'foo': 1, 'bar': 2}, {'foo': 3, 'baz': 1}]
>>> X = v.fit_transform(D)
>>> support = SelectKBest(chi2, k=2).fit(X, [0, 1])
>>> v.get_feature_names_out()
array(['bar', 'baz', 'foo'], ...)
>>> v.restrict(support.get_support())
DictVectorizer()
>>> v.get_feature_names_out()
array(['bar', 'foo'], ...)
set_output(*, transform=None)[source]#

设置输出容器。

参见 Introducing the set_output API,了解如何使用该 API 的示例。

参数:
transform{"default", "pandas", "polars"},默认为 None

配置 transformfit_transform 的输出。

  • "default":转换器的默认输出格式

  • "pandas":DataFrame 输出

  • "polars":Polars 输出

  • None:转换配置保持不变

版本 1.4 中新增: "polars" 选项已添加。

返回值:
self估计器实例

估计器实例。

set_params(**params)[source]#

设置此估计器的参数。

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

参数:
**params字典

估计器参数。

返回值:
self估计器实例

估计器实例。

transform(X)[source]#

将特征->值字典转换为数组或稀疏矩阵。

在拟合或fit_transform期间未遇到的命名特征将被静默忽略。

参数:
X形状为 (n_samples,) 的映射或映射的可迭代对象

从特征名称(任意Python对象)到特征值(字符串或可转换为dtype)的字典或映射。

返回值:
Xa{数组,稀疏矩阵}

特征向量;始终为二维。