SHARE
TWEET

Untitled

a guest Jul 24th, 2019 72 Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
  1. const model = tf.sequential();
  2.  
  3. // Add layers to the model
  4. model.add(tf.layers.dense({
  5.     inputShape: [512],
  6.     activation: 'sigmoid',
  7.     units: 2,
  8. }));
  9.  
  10. model.add(tf.layers.dense({
  11.     inputShape: [2],
  12.     activation: 'sigmoid',
  13.     units: 2,
  14. }));
  15.  
  16. model.add(tf.layers.dense({
  17.     inputShape: [2],
  18.     activation: 'sigmoid',
  19.     units: 2,
  20. }));
  21.  
  22. // Compile the model
  23. model.compile({
  24.     loss: 'meanSquaredError',
  25.     optimizer: tf.train.adam(.06), // This is a standard compile config
  26. });
RAW Paste Data
We use cookies for various purposes including analytics. By continuing to use Pastebin, you agree to our use of cookies as described in the Cookies Policy. OK, I Understand
 
Top