This code contains the PINN, PINN-SC, and PINN-Proj models used in the above paper. The code for the PINN and PINN-Proj models are contained in the '* Inference.ipynb' files. The projection layer can be enabled/disabled by uncommenting/commenting code located in the 'net_u' object. More info can be found in the comments. 'The code for the PINN-SC models are contained in the '* Inference soft constraint.ipynb' files.
Feel free to open an issue if you find a bug or have a question!
@article{baez2024guaranteeing, title={Guaranteeing Conservation Laws with Projection in Physics-Informed Neural Networks}, author={Baez, Anthony and Zhang, Wang and Ma, Ziwen and Das, Subhro and Nguyen, Lam M and Daniel, Luca}, journal={arXiv preprint arXiv:2410.17445}, year={2024} }