implicit_ODE_HL_ODE.py 文件源码

def implicit_black_box(propensities,V,X,w,h,deter_vector,stoc_positions, positions, valid,deriv):

    from scipy.integrate import ode
    #pdb.set_trace()
    deter_ode = ode(f).set_integrator('lsoda',method='bdf', with_jacobian=False)
    deter_ode.set_initial_value(X[deter_vector,:].flatten(), 0).set_f_params([propensities,V,X,deter_vector,stoc_positions, positions, valid,w])

    #pdb.set_trace()
    while deter_ode.successful() and deter_ode.t < h:
        deter_ode.integrate(h)

    #print("Black Box: \n"+ str(deter_ode.y))

    #print("iterator : \n:"+str(next_X[deter_vector,:]))
    X[deter_vector,:] = deter_ode.y.reshape(( np.sum(deter_vector),X.shape[1]))

    return X