在图层模型上使用自定义渐变tensorflow.js

时间:2019-07-26 02:53:39

标签: javascript tensorflow machine-learning tensorflow.js

我正在尝试使用Layers API轻松定义神经网络,但是我希望能够提供自定义渐变。通常,将fit method与“ x”和“ y”输入一起使用。但是,我没有对带标签的数据进行简单分类,没有“ y”。相反,我有一个自定义过程,可以根据预测来计算梯度下降的梯度。优化程序实例具有applyGradients()函数,我可以使用该函数来训练自定义渐变(此函数不在在线文档中,但在源代码中)。我认为将图层模型与applyGradients结合起来应该可以满足我的需求。

由于我仅尝试将Layers API用于NN的构造,所以我从未使用compile method,它使用了损失函数和优化器。

现在,以这种方式使用TF时遇到了问题。我无法使用model.dipose()处理该模型,因为未定义“优化器”的错误,我想是因为我从未编译过该模型。我担心会遇到其他问题,例如当我需要保存/加载模型时。

使用tensorflow.js设计我想要的最好的方法是什么?

0 个答案:

没有答案