Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add example IPython notebook for 2+1 dim Navier-Stokes problem #2

Draft
wants to merge 7 commits into
base: mogab/add-eg-nb-darcy
Choose a base branch
from

Conversation

m4e7
Copy link
Owner

@m4e7 m4e7 commented Aug 4, 2023

Migrate the example from fourier_2d_time.py from the old master branch to main:examples as fourier_2d_time.ipynb [1]. This uses the old, temporal NS data (ex. ns_V1e-3_N5_000_T50) reformatted as .pth files.

Add a helper function to datasets.navier_stokes to handle data with a temporal dimension - namely load_navier_stokes_temporal_pt.

[1] TODO

NOTE: this PR is based on top of #1. It will be rebased on neuraloperator:main when the former PR is merged.

Add utility function `datasets.load_navier_stokes_temporal_pt` to load a Navier-
Stokes dataset (from an arbitrary file location) with input/output functions
like `a(t=t0, x, y) -> u(t, x, y)` - that is, where the output functions are
time-dependent. Immediately, this facilitates loading training/testing data for
the old `fourier_2d_time.py` example.

Additionally, refactor the creation of a `UnitGaussianNormalizer` to its own
helper function.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
1 participant