Function compileModel

quantization/model_mnist.js:55–62

(model)

}

export function compileModel(model) {
  const optimizer = 'rmsprop';
  model.compile({
    optimizer: optimizer,
    loss: 'categoricalCrossentropy',
    metrics: ['accuracy'],
  });
}