如何仅执行scikit-learn管道的特定部分?

以下是与问题相关的一部分代码。如果需要完整代码,这里有一个可以下载数据的完整可重现代码:https://github.com/ageron/handson-ml2/blob/master/02_end_to_end_machine_learning_project.ipynb

我有一个管道:

prepare_select_and_predict_pipeline = Pipeline([    ('preparation', full_pipeline),    ('feature_selection', TopFeatureSelector(feature_importances, k)),    ('svm_reg', SVR(**rnd_search.best_params_))])

现在,我只想执行上述管道中的这部分:

('preparation', full_pipeline),('feature_selection', TopFeatureSelector(feature_importances, k)),

我尝试了prepare_select_and_predict_pipeline.fit(housing, housing_labels),但它也执行了SVM部分。

最终,我需要从上述管道中获得与下面代码执行相同的结果:

preparation_and_feature_selection_pipeline = Pipeline([    ('preparation', full_pipeline),    ('feature_selection', TopFeatureSelector(feature_importances, k))])housing_prepared_top_k_features = preparation_and_feature_selection_pipeline.fit_transform(housing)

我该如何做到这一点?


回答:

FeatureUnion可以解决这个问题:

from sklearn.pipeline import FeatureUnion, Pipelineprepare_select_pipeline = Pipeline([    ('preparation', full_pipeline),    ('feature_selection', TopFeatureSelector(feature_importances, k))])feats = FeatureUnion([('prepare_and_select', prepare_select_pipeline)])prepare_select_and_predict_pipeline = Pipeline([('feats', feats),                               ('svm_reg', SVR(**rnd_search.best_params_))])

您可以在深入了解Sklearn管道中找到更多相关信息

Related Posts

神经网络反向传播代码不工作

我需要编写一个简单的由1个输出节点、1个包含3个节点的…

值错误:y 包含先前未见过的标签:

我使用了 决策树分类器,我想将我的 输入 作为 字符串…

使用不平衡数据集进行特征选择时遇到的问题

我正在使用不平衡数据集(54:38:7%)进行特征选择…

广义随机森林/因果森林在Python上的应用

我在寻找Python上的广义随机森林/因果森林算法,但…

如何用PyTorch仅用标量损失来训练神经网络?

假设我们有一个神经网络,我们希望它能根据输入预测三个值…

什么是RNN中间隐藏状态的良好用途?

我已经以三种不同的方式使用了RNN/LSTM: 多对多…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注