MLproject

# vim: ft=yaml

name: gz21_ocean_momentum

entry_points:

  train:
    parameters:
      forcing_data_path: {type: string, default: None}
      run_id : {type: string, default: None}
      batchsize : {type : float, default : 8}
      learning_rate : {type : string, default : 0\1e-3}
      n_epochs : {type : float, default : 100}
      train_split : {type : float, default : 0.8}
      test_split : {type : float, default : 0.8}
      time_indices : {type : string, default : 0}
      print_every : {type : float, default : 20}
      weight_decay : {type : float, default : 0.01}
      model_module_name : {type : string, default : models.models1}
      model_cls_name : {type : string, default : FullyCNN}
      loss_cls_name : {type : string, default : HeteroskedasticGaussianLossV2}
      transformation_cls_name : {type : string, default : SquareTransform}
      submodel : {type: string, default : transform3}
      features_transform_cls_name : {type : string, default : None}
      targets_transform_cls_name : {type : string, default : None}
      subdomains_file: {type: string, default: resources/cli-configs/train-subdomains-paper.yaml}
    command: "python src/gz21_ocean_momentum/trainScript.py
        --run-id {run_id}
        --forcing-data-path {forcing_data_path}
        --subdomains-file {subdomains_file}
        --batchsize {batchsize} --learning_rate {learning_rate}
        --n_epochs {n_epochs} --train_split {train_split}
        --test_split {test_split} --time_indices {time_indices}
        --printevery {print_every} --weight_decay {weight_decay}
        --model_module_name {model_module_name}
        --model_cls_name {model_cls_name}
        --loss_cls_name {loss_cls_name}
        --transformation_cls_name {transformation_cls_name}
        --submodel {submodel}
        --features_transform_cls_name {features_transform_cls_name}
        --targets_transform_cls_name {targets_transform_cls_name}"