NeuroDiffGym / neurodiffeq

A library for solving differential equations using neural networks based on PyTorch, used by multiple research groups around the world, including at Harvard IACS.
http://pypi.org/project/neurodiffeq/
MIT License
702 stars 90 forks source link

Tidying up codes in spherical coordinates #55

Closed shuheng-liu closed 4 years ago

shuheng-liu commented 4 years ago
  1. Rewrote MonitorSpherical to plot graphs more aesthetically. Note the contour map only works for matplotlib >= v3.3 due to this issue. image

  2. Refactored functions solve_spherical and solve_spherical_system into a SphericalSolver class. These two functions, now deprecated, instantiate under the hood a SphericalSolver instance, fit it, and return the solution, loss history, etc. The solver has a lot of advantages over a solver function.

  3. Deprecated argument shuffle=True for the solver functions/class, which should be implemented in ExampleGenerators.

  4. Rewrote a bunch of docstrings. Hopefully, they make more sense.