Tensorflow与Scikit Pipeline的兼容性

时间:2019-01-08 10:28:38

标签: tensorflow scikit-learn openapi

我需要将OpenAPI中的Scikit预处理管道与Tensorflow作为学习后端而不是Scikit学习后端结合起来。 Scikit Pipelearn是否支持使用Tensorflow的后端?

1 个答案:

答案 0 :(得分:1)

最终,我们实现了一个自定义scikit估计器,该函数在fit函数上构建了Keras模型。这样,我们可以根据管道来构建具有自定义尺寸的Keras模型。