我正在尝试使用Layers API轻松定义神经网络,但是我希望能够提供自定义渐变。通常,将fit method与“ x”和“ y”输入一起使用。但是,我没有对带标签的数据进行简单分类,没有“ y”。相反,我有一个自定义过程,可以根据预测来计算梯度下降的梯度。优化程序实例具有applyGradients()函数,我可以使用该函数来训练自定义渐变(此函数不在在线文档中,但在源代码中)。我认为将图层模型与applyGradients结合起来应该可以满足我的需求。
由于我仅尝试将Layers API用于NN的构造,所以我从未使用compile method,它使用了损失函数和优化器。
现在,以这种方式使用TF时遇到了问题。我无法使用model.dipose()处理该模型,因为未定义“优化器”的错误,我想是因为我从未编译过该模型。我担心会遇到其他问题,例如当我需要保存/加载模型时。
使用tensorflow.js设计我想要的最好的方法是什么?