Unknown Loss in example Code

BeTomorrow / ReImproveJS

A framework using TensorFlow.js for Deep Reinforcement Learning

MIT License

123 stars 16 forks source link

Unknown Loss in example Code #6

Closed Sebastian-Schuchmann closed 5 years ago

Sebastian-Schuchmann commented 6 years ago

The Loss function in your example code does not exist anymore:

You should change this line:

// Finally compile the model, we also exactly use tfjs's optimizers and loss functions
// (So feel free to choose one among tfjs's)
model.compile({loss: 'crossEntropy', optimizer: 'sgd'})

to something like model.compile({loss: 'meanSquaredError', optimizer: 'sgd'})

Pravez commented 5 years ago

Indeed, done and thank you !