Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

MAINT/ENH: add test for comparing classic ode and default ode_mlp structures #353

Closed
nghi-truyen opened this issue Nov 20, 2024 · 0 comments · Fixed by #358
Closed

MAINT/ENH: add test for comparing classic ode and default ode_mlp structures #353

nghi-truyen opened this issue Nov 20, 2024 · 0 comments · Fixed by #358
Assignees
Labels
enhancement New feature or request maintenance Maintenance

Comments

@nghi-truyen
Copy link
Member

Add test for comparing classic ode and default ode_mlp structures once implicit Euler method is implemented (i.e., after #316).

def test_forward_run_mlp():
    ncpu = min(5, max(1, os.cpu_count() - 1))

    # % Test some multi layer perceptron models versus classical models
    # % TODO: add ode and ode_mlp once implicit Euler for neural ode is implemented
    mlp_to_cls_structure = {
        "zero-gr4_mlp-lr": "zero-gr4-lr",
        "zero-gr5_mlp-lr": "zero-gr5-lr",
        "zero-gr6_mlp-lr": "zero-gr6-lr",
        "zero-grd_mlp-lr": "zero-grd-lr",
        "zero-grc_mlp-lr": "zero-grc-lr",
        "zero-loieau_mlp-lr": "zero-loieau-lr",
    }
@nghi-truyen nghi-truyen added enhancement New feature or request maintenance Maintenance labels Nov 20, 2024
@nghi-truyen nghi-truyen added this to the Release v1.1.0 milestone Nov 20, 2024
@nghi-truyen nghi-truyen self-assigned this Nov 20, 2024
@nghi-truyen nghi-truyen linked a pull request Dec 7, 2024 that will close this issue
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
enhancement New feature or request maintenance Maintenance
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant