Function parseArgs

quantization/train_housing.js:30–67

()

let tf;

function parseArgs() {
  const parser = new argparse.ArgumentParser({
    description: 'TensorFlow.js Quantization Example: Training an MLP for the ' +
    'California Housing Price dataset.',
    addHelp: true
  });
  parser.addArgument('--epochs', {
    type: 'int',
    defaultValue: 200,
    help: 'Number of epochs to train the model for.'
  });
  parser.addArgument('--batchSize', {
    type: 'int',
    defaultValue: 128,
    help: 'Batch size to be used during model training.'
  });
  parser.addArgument('--validationSplit', {
    type: 'float',
    defaultValue: 0.2,
    help: 'Validation split used for training.'
  });
  parser.addArgument('--evaluationSplit', {
    type: 'float',
    defaultValue: 0.1,
    help: 'Validation split used for testing after training (evaluation).'
  });
  parser.addArgument('--modelSavePath', {
    type: 'string',
    defaultValue: './models/housing/original',
    help: 'Path to which the model will be saved after training.'
  });
  parser.addArgument('--gpu', {
    action: 'storeTrue',
    help: 'Use tfjs-node-gpu for training (requires CUDA-enabled ' +
    'GPU and supporting drivers and libraries.'
  });
  return parser.parseArgs();
}

async function main() {
  const args = parseArgs();