Function trainModelUsingFitDatasetHandler

data-generator/index.js:181–192

* Constructs a new model and trains it on a dataset pipeline built off of * GAME_GENERATOR_DATASET. The dataset pipeline performs feature calculation * and batching. * @see trainModelUsingFitDataset for training details.

()

 * @see trainModelUsingFitDataset for training details.
 */
async function trainModelUsingFitDatasetHandler() {
  STOP_REQUESTED = false;
  const model = createDNNModel();
  model.compile({
    optimizer: 'rmsprop',
    loss: 'binaryCrossentropy',
    metrics: ['accuracy'],
  });
  const dataset = GAME_GENERATOR_DATASET.map(gameToFeaturesAndLabel)
                      .batch(ui.getBatchSize());
  trainModelUsingFitDataset(model, dataset);
}

/**
 * Applies the model to the manually entered hand value and updates the UI with