Thanks to visit codestin.com
Credit goes to github.com

Skip to content

This repository contains the code and data for the SAMWO, model used in the final project of MAS557, KAIST

Notifications You must be signed in to change notification settings

lovekrand/MAS557_final

Repository files navigation

MAS557, KAIST 2024 (SPRING) Final Project

Group2, Dept. of Mathematical Sciences, Sungjong Lee, Jingu Hwang,

We modified the Multi-wavelet Operator[Gupta, et al, 2021] and designated it as Self-Adaptive Multiwavelet Operator. More experiment details are provided in the main.ipynb file.

Requirements

The code package is developed using Python 3.9 and Pytorch 2.2 with cuda 12.3.

Disclaimer

The skeleton codes and baseline model are extracted from (https://github.com/gaurav71531/mwt-operator)

image

Experiments

Data

We have provided the entire pipeline code via main.ipynb file. You can simulate the figures appearing in the final paper by downloading one of the datasets provided below. In fact, Kdv and Burger's equation uses the same dataset as in the original MWO paper. The results appearing on the pipeline code learn Burger's equation with lesser training data(which was our last experiment setup), so some of the figures may seem disparate from the one on the paper. However, if you try with KdV data, you will be able to obtain the same results.

For KdV equation : A sample generated dataset for KdV is uploaded at KdV data.

For Burgers equation : For the experiments on Burgers, the code package uses the datasets as provided in the following repository by the Authors Zongyi Li et al. We used Burgers_R10.mat data.

PDE datasets

For Wave equation : To use the wave equation as data, run the ipynb file named wave_eqn_generate.ipynb and save the .npz file.

Figures

You can check the experiment details, outcomes and figures of each PDE equation in the respective folders. We provided several figures and plots to verify our experiments were geniue.

About

This repository contains the code and data for the SAMWO, model used in the final project of MAS557, KAIST

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published