Grid search hptuning_config.yml for GCP AI platform hypertune, recommend to bayesian optimization instead,

hptuning_config_gridsearch.yml

trainingInput:
  scaleTier: CUSTOM
  masterType: n1-standard-4
  hyperparameters:
    goal: MAXIMIZE
    maxTrials: 90
    maxParallelTrials: 8
    algorithm: GRID_SEARCH
    hyperparameterMetricTag: val_accuracy
    params:
      - parameterName: neurons_num
        type: DISCRETE
        discreteValues: [50, 75, 100, 125, 150]
      - parameterName: epochs
        type: DISCRETE
        discreteValues: [30, 40, 50]
      - parameterName: batch_size
        type: DISCRETE
        discreteValues: [64, 128, 256]
      - parameterName: optimizer
        type: CATEGORICAL
        categoricalValues: ["RMSprop" , "Adam"]