const model = tf.sequential();

model.add(
  tf.layers.dense({
    units: 12,
    activation: "relu",
    inputShape: [featureCount]
  })
);

model.add(
  tf.layers.dense({
    units: 2,
    activation: "softmax"
  })
);