Function onTrainBatchEnd

lstm-text-generation/ui.js:86–97

(logs, progress, examplesPerSec)

 *   per second.
 */
export function onTrainBatchEnd(logs, progress, examplesPerSec) {
  logStatus(
      `Model training: ${(progress * 1e2).toFixed(1)}% complete... ` +
      `(${examplesPerSec.toFixed(0)} examples/s)`);
  batchLossValues.push(logs);
  const container = document.getElementById('batch-loss-canvas');
  tfvis.show.history(container, batchLossValues, ['loss'], {
    height: 300,
    zoomToFit: true,
    xLabel: 'Batch',
  });
}

export function onTrainEpochEnd(logs) {
  epochLossValues.push(logs);