OneScience-Group/Kolmogorov_flow_2d
Kolmogorov Flow 2D Dataset Description This dataset contains time series of single-channel vorticity fields obtained from numerical simulations of two-dimensional Kolmogorov Flow. It can be used for turbulence time-series forecasting, neural operator training, partial differential equation surrogate modeling, and long-horizon autoregressive forecasting. The data describes scalar vorticity fields over a two-dimensional periodic domain. The dataset contains 120… See the full description on the dataset page: https://huggingface.co/datasets/OneScience-Group/Kolmogorov_flow_2d.
<p align="center"> <strong> <span style="font-size: 30px;">Kolmogorov Flow 2D</span> </strong> </p>
Dataset Description
This dataset contains time series of single-channel vorticity fields obtained from numerical simulations of two-dimensional Kolmogorov Flow. It can be used for turbulence time-series forecasting, neural operator training, partial differential equation surrogate modeling, and long-horizon autoregressive forecasting.
The data describes scalar vorticity fields over a two-dimensional periodic domain. The dataset contains 120 trajectories, each with 320 frames, a spatial resolution of 256 x 256, a data type of float32, and a Reynolds number of Re = 1000. The dataset has been adapted for OneScience-Group/FactFormer.
Supported Tasks
Dataset Format and Structure
The main data file is in NumPy .npy format:
kf_2d_re1000_256_120seed.npyThe array shape is [120, 320, 256, 256], with the dimensions representing trajectory, time, x-grid, and y-grid, respectively. The original file is approximately 9.38 GiB. On-demand access via numpy.load(..., mmap_mode="r") is recommended to avoid copying the entire array into memory.
The data contains only vorticity fields; it does not include velocity, pressure, forcing fields, or physical time-step information.
How to Use the Dataset
Download the dataset:
hf download --dataset OneScience-Group/Kolmogorov_flow_2d --local-dir ./Kolmogorov_flow_2dTo use the dataset with FactFormer, set the data directory in FactFormer/conf/config.yaml to the download directory, and run:
cd FactFormer
python scripts/train.py
python scripts/inference.pyBy default, FactFormer downsamples the spatial resolution to 128 x 128 and uses the first 10 frames to predict the subsequent 16 frames.
Official OneScience Information
Citation and License
- Recommended model:
OneScience/FactFormer. - FactFormer: Liu-Schiaffini et al., FactFormer: Factorized Transformer for Modeling Long-Range Dependencies in PDE Surrogate Modeling.
- The documentation and supporting scripts in this repository are licensed under Apache-2.0. Before publicly distributing or redistributing the numerical data, verify its upstream source and licensing requirements.
