Keras保存模型和负载模型改变预测的形状

时间:2018-05-15 07:56:06

标签: python tensorflow scikit-learn keras

我想保存keras.wrappers.scikit_learn.KerasClassifier。我使用了keras推荐的model.save方法,但是当我加载模型时模型是keras.models.Sequential而不是keras.wrappers.scikit_learn.KerasClassifier.

这是一个问题,因为当我想进行预测时,我有一个具有这种形状的输出:

array([[0.21577413, 0.37268943, 0.13626885, 0.2752676 ],
       [0.22569732, 0.34428614, 0.15705906, 0.2729574 ],
       [0.24492404, 0.27987108, 0.21357943, 0.2616254 ],
       [0.24831992, 0.26592442, 0.22779736, 0.25795835],
       [0.24938227, 0.26125908, 0.23273101, 0.25662765]], dtype=float32)

而不是我想要的输出

array([3, 3, 3, 3, 3], dtype=int64)

所以我的问题是,如何保存和加载keras.wrappers.scikit_learn.KerasClassifier?也许还有另一个问题是改变输出的预测形状?

由于

0 个答案:

没有答案