我有一个以.pkl格式保存的梯度提升模型。我必须在tensorflowjs中加载此模型。我可以看到有一种加载Keras模型的方法,但是我找不到一种加载sklearn模型的方法。可以这样做吗?
答案 0 :(得分:0)
无法在tensorflow.js中加载sklearn模型。 Tensorflow.js允许加载以tensorflow编写的模型。
尽管,我还没有尝试过,但是我认为您可以使用scikit learn wrapper来在tensorflow中重写分类器。该模型可以保存并转换为可以在tensorflow.js中加载的格式。