DiffEqML / torchdyn

A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods
https://torchdyn.org
Apache License 2.0
1.33k stars 124 forks source link

Neuralsde #166

Closed JuanKo96 closed 7 months ago

JuanKo96 commented 1 year ago

Opening the first PR for NeuralSDE for integrating sdeint into torchdyn. Also, as I formatted code files I have made changes using black, we can use this format for future uses if everyone likes it.

The goal is to substitute sdeint module from torchsde into torchdyn native version. Will open different PRs for Brownian Interval and adjoint methods(+ other methods if possible).

codecov-commenter commented 8 months ago

Codecov Report

Attention: 74 lines in your changes are missing coverage. Please review.

Comparison is base (c708c4c) 64.59% compared to head (0bbdd3e) 66.15%.

Files Patch % Lines
torchdyn/core/neuralde.py 54.90% 23 Missing :warning:
torchdyn/core/problems.py 61.22% 19 Missing :warning:
torchdyn/numerics/sdeint.py 78.12% 14 Missing :warning:
torchdyn/core/defunc.py 70.45% 13 Missing :warning:
torchdyn/numerics/solvers/sde.py 93.97% 5 Missing :warning:

:exclamation: Your organization needs to install the Codecov GitHub app to enable full functionality.

Additional details and impacted files ```diff @@ Coverage Diff @@ ## master #166 +/- ## ========================================== + Coverage 64.59% 66.15% +1.55% ========================================== Files 27 29 +2 Lines 1881 2074 +193 ========================================== + Hits 1215 1372 +157 - Misses 666 702 +36 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.

JuanKo96 commented 7 months ago

This PR is ready to be merged.