我正在制作一个用于情感分类的RNN,并使用多对一的结构。为了让我的RNN能够在HTML文件中运行。
为了简短明了地提出问题:
Tensorflow.js中对应于Tensorflow(Python版本)的
tf.train.GradientDescentOptimizer(1.0).minimize(loss)
的等价代码是什么?
回答:
通过梯度下降,你可能会更倾向于使用随机梯度下降(随机抽样批次),其代码看起来像这样:
tf.train.stg(learningRate).minimize(loss)
更多信息请阅读这里:https://js.tensorflow.org/api/latest/#tf.train.Optimizer.minimize