MLproject

name: cmip2_6

# conda_env: conda.yaml

entry_points:
  main:
    parameters:
      ntimes : {type: float, default: 10000}
      CO2: {type: float, default: 0}
      lat_min : float
      lat_max : float
      long_min : float
      long_max : float
      factor: {type: float, default: 0}
      chunk_size: {type: string, default: 50}
      global: {type: str, default: 0}
    command: "python cmip26.py {lat_min} {lat_max} {long_min} {long_max} --CO2 {CO2} --ntimes {ntimes} --factor {factor} --chunk_size {chunk_size} --global_ {global}" 

  train:
    parameters:
      exp_id : {type: float, default: 0}
      run_id : {type: string}
      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}
    command: "python trainScript.py {exp_id} {run_id} --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}"