python - 为什么我应该在 Scikit-Learn 管道中子类化 BaseEstimator？

Question

在 scikit-learn 文档中，他们给出了 examples自定义 Transformer 的子类 BaseEstimator 和 TransformerMixin 类。我想知道，为什么在这些示例中使用了 BaseEstimator 子类？

为了尝试回答这个问题，我将它从 ItemSelector 类中取出，但 Python 没有提示。

Answer 1

BaseEstimator提供了 get_params 和 set_params 方法的实现。为什么需要这个？它可用于制作适用于 GirdSearchCV 的模型.这确保它在放置在管道中时表现良好。这只是 BaseEstimator

的应用之一

在您提供的示例中，没有执行网格搜索，这就是不需要它的原因。它包含在大多数地方(我相信)用于最佳实践，本质上是在未来验证您的代码以将网格搜索插入管道。