Skip to content

Instantly share code, notes, and snippets.

@curiousily
Created July 16, 2019 19:51
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save curiousily/5688ecd5f9223533cd0a84afef040ab0 to your computer and use it in GitHub Desktop.
Save curiousily/5688ecd5f9223533cd0a84afef040ab0 to your computer and use it in GitHub Desktop.
const model = tf.sequential();
model.add(
tf.layers.dense({
inputShape: [2],
units: 3,
activation: "relu"
})
);
model.add(
tf.layers.dense({
units: 2,
activation: "softmax"
})
);
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment