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