// The Coding Train / Daniel Shiffman

// https://thecodingtrain.com/Courses/ml5-beginners-guide/7.2-pose-classification.html

// https://youtu.be/FYgYyq-xqAw

​

// All code: https://editor.p5js.org/codingtrain/sketches/JoZl-QRPK

​

// Separated into three sketches

// 1: Data Collection: https://editor.p5js.org/codingtrain/sketches/kTM0Gm-1q

// 2: Model Training: https://editor.p5js.org/codingtrain/sketches/-Ywq20rM9

// 3: Model Deployment: https://editor.p5js.org/codingtrain/sketches/c5sDNr8eM

​

let brain;

​

function setup() {

  createCanvas(640, 480);

  let options = {

    inputs: 34,

    outputs: 6,

    task: 'classification',

    debug: true

  }

  brain = ml5.neuralNetwork(options);

  brain.loadData('gestures-new.json', dataReady);

}

​

function dataReady() {

  brain.normalizeData();

  brain.train({epochs: 50}, finished); 

}

​

function finished() {

  console.log('model trained');

  brain.save();

}