CODEWITHSUNDEEP
javascriptpythontensorflowartificial-intelligencetensorflow.js
jr123456jr987654321
jr123456jr987654321

Reputation: 324

Backpropagation in Tensorflow.js

I am making an RNN for sentiment classification while using a many to one structure. In order to make my RNN be able to run within an HTML file.

To make the question short and simple:

What is the Tensorflow.js equivalent of Tensorflow's (the python version) tf.train.GradientDescentOptimizer(1.0).minimize(loss)?

Upvotes: 1

Views: 131

Answers (1)

Isaak Eriksson
Isaak Eriksson

Reputation: 653

By gradient descent, you probably would prefer stochastic gradient descent (sampling random batches) and it would look like:

tf.train.stg(learningRate).minimize(loss)

Read more here: https://js.tensorflow.org/api/latest/#tf.train.Optimizer.minimize

Upvotes: 1

Related Questions