Skip to content

deep_macrofin.pde_model_time_step

This is a subclass of PDEModel that implements the time stepping scheme with neural network. It has exactly the same API as the base class.

PDEModelTimeStep

class PDEModelTimeStep(PDEModel):
'''
PDEModelTimeStep uses time stepping scheme + neural network to solve for optimality

PDEModel class to assign variables, equations & constraints, etc.

Also initialize the neural network architectures for each agent/endogenous variables 
with some config dictionary.
'''

set_initial_guess

def set_initial_guess(self, initial_guess: Dict[str, float])

Set the initial guess (uniform value across the state variable domain) for agents or endogenous variables. This is the boundary condition at \(t=T\) in the first time iteration.