Python sklearn DictVectorizer.restrict用法及代码示例

本文简要介绍python语言中 sklearn.feature_extraction.DictVectorizer.restrict 的用法。

用法:

restrict(support, indices=False)

使用函数选择将函数限制为支持的函数。

此函数就地修改估计器。

参数：

support：类数组

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

indices：布尔，默认=假

支持是否是索引列表。

返回：

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'], ...)