Python将数组保存为mat文件

最新推荐文章于 2024-06-24 20:04:39 发布

原创最新推荐文章于 2024-06-24 20:04:39 发布 · 4.4k 阅读

8 ·

CC 4.0 BY-SA版权

Python应用专栏收录该内容

15 篇文章

订阅专栏

这段代码将训练和验证历史记录（如损失值）保存到MAT文件中，使用scipy.io.savemat函数，确保数据转换为numpy数组格式。

import scipy.io as io

trainHist, valHist = training.trainNetwork(...)

# 要保存的变量应该是numpy array
# 如果不是，比如保存的时候报错为“list”
# 则要先转换为numpy array
# data = np.array(data)

mat_path = '../losssave/train_loss_N2V.mat'
io.savemat(mat_path, {'train_loss': trainHist})
mat_path = '../losssave/val_loss_N2V.mat'
io.savemat(mat_path, {'val_loss': valHist})