Skip to content

Commit

Permalink
Merge pull request #7 from Sebastian-Schuchmann/master
Browse files Browse the repository at this point in the history
Fixed errors in example code
  • Loading branch information
Paul Breton authored Jul 20, 2018
2 parents d6051b0 2095460 commit c396de0
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -71,7 71,7 @@ const model = new ReImprove.Model.FromNetwork(network, modelFitConfig);

// 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'})
model.compile({loss: 'meanSquaredError', optimizer: 'sgd'})

```

Expand Down Expand Up @@ -128,7 128,7 @@ async function step(time) {

let inputs = getInputs(); // Need to give a number[] of your inputs for one teacher.
await academy.step([ // Let the magic operate ...
{teacherName: teacher, agentInputs: inputs}
{teacherName: teacher, agentsInput: inputs}
]);

}
Expand Down

0 comments on commit c396de0

Please sign in to comment.