Function createModel

abalone-node/model.js:26–42

* Builds and returns Multi Layer Perceptron Regression Model. * * @param {number} inputShape The input shape of the model. * @returns {tf.Sequential} The multi layer perceptron regression mode l.

(inputShape)

 * @returns {tf.Sequential} The multi layer perceptron regression mode  l.
 */
function createModel(inputShape) {
  const model = tf.sequential();
  model.add(tf.layers.dense({
    inputShape: inputShape,
    activation: 'sigmoid',
    units: 50,
  }));
  model.add(tf.layers.dense({
    activation: 'sigmoid',
    units: 50,
  }));
  model.add(tf.layers.dense({
    units: 1,
  }));
  model.compile({optimizer: tf.train.sgd(0.01), loss: 'meanSquaredError'});
  return model;
}

module.exports = createModel;