mindearth.data.Era5Data

class mindearth.data.Era5Data(data_params, run_mode='train')

Era5Data类通过MindSpore框架处理ERA5数据集生成数据生成器。Era5Data类继承了Data类。

参数：

• data_params (dict) - 模型中的相关数据参数。

• run_mode (str, 可选) - 决定数据集用于训练、验证还是测试。支持 'train', 'test', 'valid'。默认值： 'train'。

• kno_patch (bool, 可选) - 决定数据集是否分割成小块。如果为 True，则假定数据已预处理，并且不会执行进一步的分割。如果为 False， 则将根据指定的参数将数据分割成小块。默认值： False。

支持平台：

Ascend GPU

样例：

>>> from mindearth.data import Era5Data
>>> data_params = {
...     'name': 'era5',
...     'root_dir': './dataset',
...     'feature_dims': 69,
...     't_in': 1,
...     't_out_train': 1,
...     't_out_valid': 20,
...     't_out_test': 20,
...     'valid_interval': 1,
...     'test_interval': 1,
...     'train_interval': 1,
...     'pred_lead_time': 6,
...     'data_frequency': 6,
...     'train_period': [2015, 2015],
...     'valid_period': [2016, 2016],
...     'test_period': [2017, 2017],
...     'patch': True,
...     'patch_size': 8,
...     'batch_size': 8,
...     'num_workers': 1,
...     'grid_resolution': 1.4,
...     'h_size': 128,
...     'w_size': 256
... }
>>> dataset_generator = Era5Data(data_params)