ntt123 commited on
Commit
e34beee
1 Parent(s): 2096407

Update index.html

Browse files
Files changed (1) hide show
  1. index.html +1 -1
index.html CHANGED
@@ -69,7 +69,7 @@
69
  if (this_.is_terminated()) return { "terminated": true, "action": -1 };
70
  const obs = tf.tensor(this_.board, [this_.num_rows, this_.num_cols], 'float32');
71
  const normalized_obs = tf.mul(obs, this_.ai_player);
72
- const [action_logits, value] = this_.agent.predict(normalized_obs);
73
  const action = await tf.argMax(action_logits).array();
74
  return {
75
  "terminated": false,
69
  if (this_.is_terminated()) return { "terminated": true, "action": -1 };
70
  const obs = tf.tensor(this_.board, [this_.num_rows, this_.num_cols], 'float32');
71
  const normalized_obs = tf.mul(obs, this_.ai_player);
72
+ const [action_logits, value] = this_.agent.execute(normalized_obs, ["output_0", "output_1"]);
73
  const action = await tf.argMax(action_logits).array();
74
  return {
75
  "terminated": false,