https://github.com/piglaker/fesnet

yeah
https://github.com/piglaker/fesnet

Last synced: 3 months ago
JSON representation

yeah

• Host: GitHub
• URL: https://github.com/piglaker/fesnet
• Owner: piglaker
• Created: 2021-05-16T07:11:40.000Z (about 4 years ago)
• Default Branch: master
• Last Pushed: 2021-06-03T17:16:02.000Z (about 4 years ago)
• Last Synced: 2025-01-16T18:57:45.987Z (5 months ago)
• Language: Jupyter Notebook
• Size: 70.8 MB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
## FesNet for solving PDE which based on finite-element in Seismic

![Alt text](/data/mirror/0_earthquake.png)

![Alt text](/data/mirror/0_acc.png)

# Model

![Alt text](/net/model.png)

# Train 

    Dataset Para:

        super_element_dim = 294,

        super_step_size = 1000,

        super_step_t = 0.5, 

        super_hot_start = 1/35, 

        super_size=31,

        path = "./data/"

        

    Train para:

        Adam,StepLR

        lr = 0.002

        hidden_dim = 400

        logits_size = 10

        num_layers = 2

        epoch = 10

        max_length = 10

        clip_threh = 0.5

        step_size = 5

        gamma = 0.5

    

![Alt text](./Loss_final.png)

# Predict 

Test_dataset:

Time Cost:  5.15300178527832 s

Time Cost:  3.7200005054473877 s

Time Cost:  3.778003692626953 s

Time Cost:  4.040002107620239 s

Time Cost:  3.366999626159668 s

Time Cost:  3.7520222663879395 s

Time Cost:  3.792003870010376 s

Time Cost:  3.3139984607696533 s

        Teacher-Forcing   Self-Regression

    MSE : 4.136940627009608, 4.143258782220073

    R2  : 0.6857                  0.6852

demo:

Perfect one:

![Alt text](/predict/7_trans_prediction.png)

![Alt text](/predict/7_trans_selfre_prediction.png)

![Alt text](/predict/7_trans_true.png)

Bad one:

![Alt text](/predict/5_trans_prediction.png)

![Alt text](/predict/5_trans_selfre_prediction.png)

![Alt text](/predict/5_trans_true.png)