Pen Settings

HTML

CSS

CSS Base

Vendor Prefixing

Add External Stylesheets/Pens

Any URL's added here will be added as <link>s in order, and before the CSS in the editor. If you link to another Pen, it will include the CSS from that Pen. If the preprocessor matches, it will attempt to combine them before processing.

+ add another resource

JavaScript

Babel is required to process package imports. If you need a different preprocessor remove all packages first.

Add External Scripts/Pens

Any URL's added here will be added as <script>s in order, and run before the JavaScript in the editor. You can use the URL of any other Pen and it will include the JavaScript from that Pen.

+ add another resource

Behavior

Save Automatically?

If active, Pens will autosave every 30 seconds after being saved once.

Auto-Updating Preview

If enabled, the preview panel updates automatically as you code. If disabled, use the "Run" button to update.

Format on Save

If enabled, your code will be formatted when you actively save your Pen. Note: your code becomes un-folded during formatting.

Editor Settings

Code Indentation

Want to change your Syntax Highlighting theme, Fonts and more?

Visit your global Editor Settings.

HTML

              
                <p>View the output in the console.</p>
<p><button id="btn_train">Train the Model</button></p>
<p>Read a full explanation of this demo on <a href="https://www.greenzeta.com/creating-a-simple-classification-neural-network-with-tensorflow-js/" target="_blank">GreenZeta.com</a></p>
<img src="https://www.greenzeta.com/wp-content/uploads/2018/05/tensorflow-js-300x300.png" alt="" width="250" height="250">
              
            
!

CSS

              
                
              
            
!

JS

              
                // Solve for XOR
const LEARNING_RATE = 0.1;
const EPOCHS = 200;

// Define the training data
const xs = [[0,0],[0,1],[1,0],[1,1]];
const ys = [0,1,1,0];

// Instantiate the training tensors
let xTrain = tf.tensor2d(xs, [4,2]);
let yTrain = tf.oneHot(tf.tensor1d(ys).toInt(), 2);

// Define the model.
const model = tf.sequential();
// Set up the network layers
model.add(tf.layers.dense({units: 5, activation: 'sigmoid', inputShape: [2]}));
model.add(tf.layers.dense({units: 2, activation: 'softmax', outputShape: [2]}));
// Define the optimizer
const optimizer = tf.train.adam(LEARNING_RATE);
// Init the model
model.compile({
    optimizer: optimizer,
    loss: 'categoricalCrossentropy',
    metrics: ['accuracy'],
});

const button = document.querySelector('#btn_train');
button.addEventListener('mouseover', evt => {
  console.log('Training... This will take a moment.');
  window.setTimeout(function(){TrainModel()},500);
});

// Put the training/prediction into a function because it was slowing page load.
let TrainModel = function(){
  // Train the model
  const history = model.fit(xTrain, yTrain, {
    epochs: EPOCHS,
    validationData: [xTrain, yTrain],
  }).then(()=>{
    // Try the model on a value
     const input = tf.tensor2d([0,1], [1, 2]);
     const predictOut = model.predict(input);
     const logits = Array.from(predictOut.dataSync());
     console.log('prediction', logits, predictOut.argMax(-1).dataSync()[0]);
  });
}

console.log('Ready. Press the Train Button.');
              
            
!
999px

Console