Function onTrainEpochEnd

lstm-text-generation/ui.js:99–107

(logs)

}

export function onTrainEpochEnd(logs) {
  epochLossValues.push(logs);
  const container = document.getElementById('epoch-loss-canvas');
  tfvis.show.history(container, epochLossValues, ['loss', 'val_loss'], {
    height: 300,
    zoomToFit: true,
    xLabel: 'Epoch',
  });
}

/**
 * A function to call when text generation begins.