metadata
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.
Getting Started
Prerequisites
- Python 3.12+
- PyTorch
- NumPy
- Matplotlib
Installation
git clone https://github.com/sr5434/pinns.git
cd pinns
pip install -r requirements.txt
Usage
Evaluate on Heat Equation
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
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)
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)
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