Function modelSummary

tfjs-vis/src/show/model.ts:55–75

(container: Drawable, model: LayersModel)

 * }
 */
export async function modelSummary(container: Drawable, model: LayersModel) {
  const drawArea = getDrawArea(container);
  const summary = getModelSummary(model);

  const headers = [
    'Layer Name',
    'Output Shape',
    '# Of Params',
    'Trainable',
  ];

  const values = summary.layers.map(
      l =>
          [l.name,
           l.outputShape,
           l.parameters,
           l.trainable,
  ]);

  table(drawArea, {headers, values});
}

/**
 * Renders summary information about a layer and a histogram of parameters in