OneScience-Group/AtmosphericDA-DModel

1{2  "format_version": "2.0",3  "model_name": "AtmosphericDA-DModel",4  "model_type": "dense-analysis-increment-correction",5  "architectures": ["DModel", "HybridSurrogate", "TwoLayerQG"],6  "framework": "PyTorch",7  "paper_framework": "TensorFlow",8  "domain": "earth-science",9  "task": "two-layer-streamfunction-model-error-correction",10  "license": "Apache-2.0",11  "implementation": {12    "entry_point": "model/atmosphericda_dmodel.py",13    "train_script": "scripts/train.py",14    "inference_script": "scripts/inference.py",15    "evaluation_script": "scripts/result.py",16    "synthetic_data_script": "scripts/fake_data.py"17  },18  "architecture": {19    "family": "D model linear Dense analysis-increment correction",20    "input_shape": ["B", 2, 20, 40],21    "output_shape": ["B", 2, 20, 40],22    "state_layout": "BCYX",23    "state_size": 1600,24    "hidden_size": 8,25    "activation": "linear",26    "target": "x_a_{k+1} - M_o(x_a_k)"27  },28  "data": {29    "protocol": "hybrid_qg_analysis_increment_npz_v2",30    "state_shape": [2, 20, 40],31    "observation_shape_per_window": [12, 50],32    "observation_operator": "bilinear at random off-grid locations",33    "observation_covariance": "0.1 I"34  },35  "configuration_sources": ["conf/config.yaml", "model/atmosphericda_dmodel.py", "scripts/fake_data.py", "scripts/train.py", "scripts/inference.py", "scripts/result.py"]36}37