evalieben / odesolver-pinns Goto Github PK

dx/dt = -x   

    ODE4 = ODE(t1, t2, nb_samples)
    ODE4.setEquation([ODE4.diff(1,1), 1],['x'],[1,1],"ODE.constant(0)")
    ODE4.addBoundary(0, 0.0, 1.00)  # x[0.0] = 1.0
    ODE4.setExact(log(ODE4.xVar()),"exp(-t)")

    solve_with_dct = {
        "network_layers": 2,   # origin: 8
        "cells": 30,  # origin: 60
        "learn_rate": 1.e-3,  # orig 1.e-5
        "weight_decay": 0.005,
        "max_steps": 40001,
        "threads": 2,
        "reduction": 'mean'
    }

    ODE4.solve(**solve_with_dct)