ClassNamePrefixFeaturesOutMixin#
- 类 sklearn.base.ClassNamePrefixFeaturesOutMixin[源]#
通过添加前缀来生成自身名称的转换器的混入类。
当转换器需要生成自己的输出特征名称时(例如
PCA
),这个混入类很有用。例如,如果PCA
输出3个特征,则生成的输出特征名称为:["pca0", "pca1", "pca2"]
。这个混入类假定在转换器拟合时定义了一个
_n_features_out
属性。_n_features_out
是转换器将在transform
或fit_transform
中返回的输出特征的数量。示例
>>> import numpy as np >>> from sklearn.base import ClassNamePrefixFeaturesOutMixin, BaseEstimator >>> class MyEstimator(ClassNamePrefixFeaturesOutMixin, BaseEstimator): ... def fit(self, X, y=None): ... self._n_features_out = X.shape[1] ... return self >>> X = np.array([[1, 2], [3, 4]]) >>> MyEstimator().fit(X).get_feature_names_out() array(['myestimator0', 'myestimator1'], dtype=object)