sr5434/PINN-Collection

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