|
|
--- |
|
|
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 |
|
|
``` |
