【问题标题】:How to save a tensor如何保存张量
【发布时间】:2021-08-30 06:24:05
【问题描述】:
我有一个包含 1000 个项目的数据集。在针对数据训练模型之前,我会对数据进行标准化。
我现在想使用该模型进行预测。但是,据我了解,我需要对将提供给我需要预测的模型的输入进行标准化。为了做到这一点,我需要在训练时计算出平均值和标准值。
虽然我可以将它打印到控制台,但如何“保存”它 - 以供以后使用?我正在尝试了解此处有关如何保存训练数据标准化时使用的均值和标准差的过程 - 以便我可以在进行预测时再次使用它。
【问题讨论】:
标签:
tensorflow.js
danfojs
【解决方案1】:
我确定我们可以先通过以下方式得到张量的数组表示:
// tensor here is the tensor variable that contains the tensor
const tensorAsArray = tensor.arraySync()
然后,我们像其他字符串一样将其保存到文件中
fs.writeFile(myFilePath, JSON.stringify(tensorAsArray), 'utf-8')
要读回它并将其用作张量,我们会做相反的事情:
const tensorAsArray = JSON.parse(fs.readFile(myFilePath, 'utf-8'))
const tensor = tf.tensor(tensorAsArray)
这让我可以保存平均值和标准以供以后使用。