maubin/ML_GravPotBCs
0
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

COCA

Browse source
This commit is contained in:
Mayeul Aubin 2025-06-26 11:31:29 +02:00
parent 066a5b3958
commit e4308c003b
3 changed files with 329 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
sCOCA_ML/dataset/momenta_dataset.py
View file

@ -40,7 +40,7 @@ class MomentaDataset(Dataset):
initial_conditions_variables:tuple|list=['DM_delta', 'DM_phi'],
target_variable:str='gravpot',
style_files:str='cosmo_and_time_parameters',
style_keys:list|None=["D1", "D2"],
style_keys:list|None=["a"],
max_time:int=100):
"""
Dataset for residual momenta data (COCA).