如何使用Pycharm中的Google colab导出的模型

Question

我有一个经过培训的LSTM Keras Tensorflow模型，并以.h5（HDF5）格式导出。我的本地机器不支持keras tensorflow。我尝试安装。但是不起作用。

因此，我使用google colabs并导出了模型。

我想知道，如何在pycharm中使用导出的模型

编辑：我刚刚在我的机器上安装了tensorflow

预先感谢

Answer 1

0
投票

您仍然需要使用keras和tensorflow才能使用模型。

Answer 2

找到答案：

我已按如下所示导出模型

model.save('/content/drive/My Drive/Colab Notebooks/model.h5')

然后我下载了文件并将其保存在其他代码所在的文件夹中。我已经安装了tensorflow。

[下一步，我加载代码并使用保存的模型进行预测，如下所示。

import keras
model=keras.models.load_model('/content/drive/My Drive/Colab Notebooks/model.h5')
model.predict(instace)