Function main

quantization/eval_housing.js:45–75

()

}

async function main() {
  const args = parseArgs();
  if (args.gpu) {
    tf = require('@tensorflow/tfjs-node-gpu');
  } else {
    tf = require('@tensorflow/tfjs-node');
  }

  const {count, featureMeans, featureStddevs, labelMean, labelStddev} =
      await getDatasetStats();

  const validationSplit = 0.2;
  const evaluationSplit = 0.1;
  const {evalXs, evalYs} =
      await getNormalizedDatasets(
          count, featureMeans, featureStddevs, labelMean, labelStddev,
          validationSplit, evaluationSplit);

  console.log(`Loading model from ${args.modelSavePath}...`);
  const model = await tf.loadLayersModel(`file://${args.modelSavePath}`);
  compileModel(model);

  console.log(`Performing evaluation...`);
  const t0 = tf.util.now();
  const evalOutput = model.evaluate(evalXs, evalYs);
  const t1 = tf.util.now();
  console.log(`\nEvaluation took ${(t1 - t0).toFixed(2)} ms.`);
  console.log(
      `\nEvaluation result:\n` +
      `  Loss = ${evalOutput.dataSync()[0].toFixed(6)}`);
}

if (require.main === module) {
  main();