作者:此人已死_0824 | 来源:互联网 | 2022-12-30 16:50
我目前正在阅读"Scikit-Learn&TensorFlow的动手机器学习".当我尝试重新创建Transformation Pipelines代码时出错.我怎样才能解决这个问题?
码:
from sklearn.pipeline import Pipeline
from sklearn.preprocessing import StandardScaler
num_pipeline = Pipeline([('imputer', Imputer(strategy = "median")),
('attribs_adder', CombinedAttributesAdder()),
('std_scaler', StandardScaler()),
])
housing_num_tr = num_pipeline.fit_transform(housing_num)
from sklearn.pipeline import FeatureUnion
num_attribs = list(housing_num)
cat_attribs = ["ocean_proximity"]
num_pipeline = Pipeline([
('selector', DataFrameSelector(num_attribs)),
('imputer', Imputer(strategy = "median")),
('attribs_adder', CombinedAttributesAdder()),
('std_scaler', StandardScaler()),
])
cat_pipeline = Pipeline([('selector', DataFrameSelector(cat_attribs)),
('label_binarizer', LabelBinarizer()),
])
full_pipeline = FeatureUnion(transformer_list = [("num_pipeline", num_pipeline),
("cat_pipeline", cat_pipeline),
])
# And we can now run the whole pipeline simply:
housing_prepared = full_pipeline.fit_transform(housing)
housing_prepared
错误:
---------------------------------------------------------------------------
NameError Traceback (most recent call last)
in ()
43
44 num_pipeline = Pipeline([
---> 45 ('selector', DataFrameSelector(num_attribs)),
46 ('imputer', Imputer(strategy = "median")),
47 ('attribs_adder', CombinedAttributesAdder()),
NameError: name 'DataFrameSelector' is not defined
Stephen Rauc..
16
DataFrameSelector
没有被发现,需要进口.它不是其中的一部分sklearn
,但sklearn功能中提供了相同名称的内容:
from sklearn_features.transformers import DataFrameSelector
(DOCS)
1> Stephen Rauc..:
DataFrameSelector
没有被发现,需要进口.它不是其中的一部分sklearn
,但sklearn功能中提供了相同名称的内容:
from sklearn_features.transformers import DataFrameSelector
(DOCS)
2> dr2509..:
from sklearn.base import BaseEstimator, TransformerMixin
class DataFrameSelector(BaseEstimator, TransformerMixin):
def __init__(self, attribute_names):
self.attribute_names=attribute_names
def fit(self, X, y=None):
return self
def transform(self, X):
return X[self.attribute_names].values
这应该工作。
有时最好参考以下来源:使用Scikit-Learn和TensorFlow进行动手机器学习第97页