more banging on the tensorflow api

Author: avindra

src/net/check_model.js

@@ -1,6 +1,12 @@
 import * as  tf from '@tensorflow/tfjs-node-gpu';
 
 const model = await tf.loadLayersModel('file:///dev/shm/brain/model.json');
-model.loadWeights()
 
-console.log('prediction', model.predict(tf.tensor([ Array(400) ])));
+// obscure looking tf operations taken from
+// https://github.com/prouhard/tfjs-mountaincar/blob/master/src/js/model.js#L69-L74
+const pred = model.predict(tf.tensor([ Array(400) ]));
+const sigmoid = tf.sigmoid(pred);
+const probs = tf.div(sigmoid, tf.sum(sigmoid));
+const end = tf.multinomial(probs, 1).dataSync()[0] - 1;
+
+console.log('probs', end);

src/net/model.js

@@ -11,5 +11,6 @@ export const createModel = (inputVectorSize) => {
 
   // Prepare the model for training: Specify the loss and the optimizer.
   model.compile({loss: 'meanSquaredError', optimizer: 'sgd'});
+  model.summary();
   return model;
 }