Hugging Face's logo

sr5434
/
PINN-Collection

PyTorch
physics
pinn
pde
heat-equation
burgers-equation
schrodinger
Eval Results (legacy)
Model card Files
xet
Community
PINN-Collection
File size: 2,929 Bytes 
0d45fb8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4e47574
 
0d45fb8
 
 
4e47574
0d45fb8
 
 
 
 
 
 
4e47574
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
0d45fb8
 
b218a13
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4e47574
 
b218a13
 
 
 
4e47574
b218a13
 
 
 
4e47574
 
 
b218a13
4e47574
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
 
---
license: mit
tags:
- physics
- pinn
- pde
- pytorch
- heat-equation
- burgers-equation
- schrodinger
model-index:
- name: Heat Equation 3D
  results:
  - task:
      type: physics-simulation
      name: Heat Equation
    dataset:
      type: synthetic
      name: Generated Data
    metrics:
    - type: mean_absolute_error
      name: Max Error
      value: 0.04
- name: Burgers' Equation 1D
  results:
  - task:
      type: physics-simulation
      name: Burgers' Equation
    dataset:
      type: synthetic
      name: Generated Data
    metrics:
    - type: mean_absolute_error
      name: Max Error
      value: 0.045
- name: Schrödinger Equation 1D
  results:
  - task:
      type: physics-simulation
      name: Schrödinger Equation
    dataset:
      type: synthetic
      name: Generated Data
    metrics:
    - type: mean_absolute_error
      name: Max Error
      value: 0.065
- name: Schrödinger Equation Hydrogen
  results:
  - task:
      type: physics-simulation
      name: Schrödinger Equation
    dataset:
      type: synthetic
      name: Generated Data
    metrics:
    - type: mean_absolute_error
      name: Max Error
      value: 0.02
---

# Physics-Informed Neural Networks
This contains the pretrained model weights for a collection of physics-informed neural networks (PINNs) that are designed to solve problems in quantum mechanics, thermodynamics, and fluid dynamics by learning directly from physics rather than from labeled data. For training and evaluation scripts, check out the [main repository](https://github.com/sr5434/pinns).

## Getting Started

### Prerequisites
- Python 3.12+
- PyTorch
- NumPy
- Matplotlib

### Installation
```bash
git clone https://github.com/sr5434/pinns.git
cd pinns
pip install -r requirements.txt
```

### Usage
#### Evaluate on Heat Equation
```bash
cd heat_equation

# Generate visualizations
curl "https://huggingface.co/sr5434/PINN-Collection/resolve/main/heat_equation_3d.pt?download=true" -o heat_equation_3d.pt
python heat_equation_visualizer_3d.py
```
#### Evaluate on Burger's Equation
```bash
cd burgers_equation

# Generate visualizations
curl "https://huggingface.co/sr5434/PINN-Collection/resolve/main/burgers_equation_1d.pt?download=true" -o burgers_equation_1d.pt
python burgers_equation_visualization_1d.py
```

#### Test Pretrained Model on Schrödinger Equation(1D)
```bash
cd schrodingers_equation

# Generate visualizations
curl -L "https://huggingface.co/sr5434/PINN-Collection/resolve/main/schrodingers_equation_1d.pt?download=true" -o schrodingers_equation_1d.pt
python schrodingers_visualization_1d.py
```


#### Test Pretrained Model on Schrödinger Equation(Hydrogen)
```bash
cd schrodingers_equation

# Generate visualizations
curl -L "https://huggingface.co/sr5434/PINN-Collection/resolve/main/schrodingers_equation_hydrogen.pt?download=true" -o schrodingers_equation_hydrogen.pt
python schrodingers_visualization_hydrogen.py
```