Python 如何在Pycharm中使用从google colab导出的模型
我有一个LSTM Keras Tensorflow模型,以.h5(HDF5)格式培训和导出。 我的本地机器不支持keras tensorflow。我试过安装。但是不起作用 因此,我使用了google colabs并导出了模型 我想知道,如何在pycharm中使用导出的模型 编辑:我刚刚在我的机器上安装了tensorflowPython 如何在Pycharm中使用从google colab导出的模型,python,tensorflow,keras,pycharm,google-colaboratory,Python,Tensorflow,Keras,Pycharm,Google Colaboratory,我有一个LSTM Keras Tensorflow模型,以.h5(HDF5)格式培训和导出。 我的本地机器不支持keras tensorflow。我试过安装。但是不起作用 因此,我使用了google colabs并导出了模型 我想知道,如何在pycharm中使用导出的模型 编辑:我刚刚在我的机器上安装了tensorflow 提前感谢您仍然需要keras和tensorflow来使用该模型。找到了答案: 我导出了模型,如下所示 model.save('/content/drive/My Drive/
提前感谢您仍然需要keras和tensorflow来使用该模型。找到了答案: 我导出了模型,如下所示
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)
接受的答案是正确的,但它忽略了您首先需要挂载“/content/drive” 然后可以保存模型的权重:
model.save_weights('my_model_weights.h5')
..甚至保存整个模型:
model.save('my_model.h5')
完成后,使用以下方法断开安装点的连接:
drive.flush_and_unmount()
您需要TensorFlow才能使用导出的模型。我现在安装了它
drive.flush_and_unmount()