let neuralNetwork;

​

function setup() {

  noCanvas();

  let nnOptions = {

    dataUrl: 'titanic.csv',

    inputs: ['fare_class','sex', 'age'],

    outputs: ['survived'],

    task: 'classification',

    debug: true,

  };

  neuralNetwork = ml5.neuralNetwork(nnOptions, modelReady);

  classTitle = createP('Class');

  classInput = createInput('first', 'text');

  sexTitle = createP('Sex');

  sexInput = createInput('female', 'text');

  ageTitle = createP('Age');

  ageInput = createInput('20', 'text');

  submitButton = createButton('Submit');

  submitButton.mousePressed(classify);

  resP = createP('results:')

}

​

function modelReady() {

  neuralNetwork.normalizeData();

  neuralNetwork.train({ epochs: 200 }, whileTraining, finishedTraining);

}

​

function whileTraining(epochs, log) {

  console.log('epochs', epochs)

  console.log('log', log.loss)

}

​

function finishedTraining() {

  classify();

}

​

function classify() {

  console.log('classify');

  const input = [classInput.value(), sexInput.value(), ageInput.value()];

  neuralNetwork.classify(input, gotResults);

}

​

function gotResults(err, res) {

  if (res) {

    resP.html('results: ' + res[0].label + ', ' + res[0].confidence)

  }

}

​

​

​

​

​

​