在 Google Colab 中下载 h5 文件

Question

经过大量搜索，我仍然找不到从远程检索 H5 文件的方法，这是我的代码：

import requests
import io
import h5py

url = 'https://raw.githubusercontent.com/aymeric75/IA/master/weights_bitcoin.h5'
r = requests.get(url, allow_redirects=True)
open('weights_btc.h5', 'wb').write(r.content)


if not h5py.is_hdf5('weights_btc.hdf5'):
    raise ValueError('Not an hdf5 file')

谢谢

p.s：我在 Google Colab 中需要这个，因为当保存与笔记本不同的文件时，退出会话时它会被删除。

返回此错误：

值错误：不是 hdf5 文件

Answer 1

这可能有帮助： 在 Google Colab 中加载 ML 数据的 4 种很棒的方法

此外，如果您想更频繁地使用该模型并且不想在每次运行时下载它，我建议您将 google 驱动器与 collab 连接。以下博客将对您有所帮助。

在 Google Colab 中打开 Google Drive 文件的最简单方法

Answer 2

我认为您正在尝试将数据集保存到 Google 协作中的 h5 文件中。实现这一目标的一种方法是

import io
import h5py

url = 'https://raw.githubusercontent.com/aymeric75/IA/master/weights_bitcoin.h5'
r = requests.get(url, allow_redirects=True)

f= h5py.File('weights_btc.h5', 'w')
f.create_dataset(name ="Test", data = r.content)
f.close()

with open('weights_btc.h5', 'wb') as f:
    f.write(r.content)

本质上，Github 上的 h5 文件由附加到一个组的数据集或数据集集合组成。我认为直接将 H5 文件下载到 H5 对象中会更 Pythonic，就像我们将远程链接下载到 pandas 中一样。不知道是否有办法做到这一点。

Answer 3

cnn.save('trained_model.h5')

此命令将用于在 Google Colab 中下载 .h5 文件