index.js

import {MnistData} from './data';
import * as model from './model';
import * as ui from './ui';

var $ = require('jquery');

let data;
async function load(chart_id) {
  ui.isLoading(chart_id);
  data = new MnistData();
  await data.load();
}

async function train(LEARNING_RATE, chart_id, noise=false, subId=undefined) {
  ui.isTraining(chart_id);
  await model.train('CNN', data, ui.trainingLog, LEARNING_RATE, chart_id, noise, subId);
}

async function train_BN(LEARNING_RATE, chart_id, noise=false, subId=undefined) {
  ui.isTraining(chart_id);
  await model.train('CNN_BN', data, ui.trainingLog, LEARNING_RATE, chart_id, noise, subId);
}

async function test() {
  const testExamples = 50;
  const batch = data.nextTestBatch(testExamples);
  const predictions = model.predict(batch.xs);
  const labels = model.classesFromLabel(batch.labels);

  ui.showTestResults(batch, predictions, labels);
}

async function train0() {
  var id = '0';
  await load(id);
  await train(0.1, id);
  await train_BN(0.1, id);
  await train(0.5, id);
  await train_BN(0.5, id);
  ui.done(id);
}

async function train1() {
  var id = '1';
  await load(id);
  await train(undefined, id, false, '_1');
  await train_BN(undefined, id, false, '_2');
  await train_BN(undefined, id, true, '_3');
  ui.done(id);
}

async function train2() {
  var id = '2';
  await load(id);
  await train(undefined, id, false);
  await train_BN(undefined, id, false);
  ui.done(id);
}



document.getElementById('train0').addEventListener('click', function() {
  train0();
});

document.getElementById('train1').addEventListener('click', function() {
  train1();
});

document.getElementById('train2').addEventListener('click', function() {
  train2();
});