MLproject

name: Generative downscaling

conda_env: environment.yml

entry_points:
  glow-jflvm:
    parameters:
      scale: {type: int, default: 4}
      layers: {type: int, default: 3}
      depth: {type: int, default: 8}
      min_filters: {type: int, default: 32}
      max_filters: {type: int, default: 256}
      lam: {type: float, default: 1.0}
      alpha: {type: float, default: 0.0}
      dnet_layers: {type: int, default: 3}
      dnet_filters: {type: int, default: 64}
      batch_size: {type: int, default: 10}
      epochs: {type: int, default: 20}
      validate_freq: {type: int, default: 10}
      prior: {type: string, default: "standard"}
      mode: {type: string, default: "cv"}
      test_size: {type: int, default: 146}
      splits: {type: int, default: 5}
      var: {type: string, default: "MAXT"}
      region: {type: string, default: "southeast_us"}
      auth: {type: string, default: "gcs.secret.json"}
      data_lr: {type: string, default: "ras/daily-1deg"}
    command: "python run_experiment.py glow-jflvm {data_lr} --scale {scale} --layers {layers} --depth {depth} \
                --min-filters {min_filters} --max-filters {max_filters} --lam {lam} --alpha {alpha} --prior {prior} \
                --dnet-layers {dnet_layers} --dnet-filters {dnet_filters} --splits {splits} --test-size {test_size} --batch-size {batch_size} \
                --var {var} --epochs {epochs} --validate-freq {validate_freq} --region {region} --mode {mode} --auth {auth}"

  bcsd:
    parameters:
      data_lr: {type: string, default="erai/daily-1deg"}
      scale: {type: int, default: 4}
      var: {type: string, default: "MAXT"}
      test_size: {type: int, default: 146}
      splits: {type: int, default: 5}
      dask_host: {type: string, default: ""}
      auth: {type: string, default: "gcs.secret.json"}
      
    command: "python run_experiment.py bcsd {data_lr} --scale {scale} --var {var} --test-size {test_size} \
                --splits {splits} --dask-host {dask_host} --auth {auth}"
    
  bmd:
    parameters:
      scale: {type: int, default: 4}
      epochs: {type: int, default: 50}
      batch_size: {type: int, default: 100}
      validate_freq: {type: int, default: 10}
      var: {type: string, default: "MAXT"}
      test_size: {type: int, default: 146}
      splits: {type: int, default: 5}
      data_lr: {type: string, default: "ras/daily-1deg"}
      auth: {type: string, default: "gcs.secret.json"}
      
    command: "python run_experiment.py bmd {data_lr} --scale {scale} --var {var} --epochs {epochs} --batch-size {batch_size} \
                  --test-size {test_size} --splits {splits} --validate-freq {validate_freq} --auth {auth}"