def test_eval_formula(): from dolo.compiler.eval_formula import eval_formula from dolo import yaml_import, approximate_controls, simulate model = yaml_import('examples/models/rbc.yaml') dr = approximate_controls(model) sim = simulate(model, dr) rr = eval_formula("delta*k-i", sim, context=model.calibration) rr = eval_formula("y(1) - y", sim, context=model.calibration) sim['diff'] = model.eval_formula("delta*k-i", sim) model.eval_formula("y(1) - y", sim) sim['ddiff'] = model.eval_formula("diff(1)-diff(-1)", sim)
def eval_formula(self, expr, dataframe=None, calib=None): from dolo.compiler.eval_formula import eval_formula if calib is None: calib = self.calibration return eval_formula(expr, dataframe=dataframe, context=calib)
def test_eval_formula(): from dolo.compiler.eval_formula import eval_formula from dolo import yaml_import, time_iteration, simulate model = yaml_import('examples/models/rbc.yaml') dr = time_iteration(model) sim = simulate(model, dr) sim = sim.sel(N=0) sim = sim.to_pandas() rr = eval_formula("delta*k-i", sim, context=model.calibration) rr = eval_formula("y(1) - y", sim, context=model.calibration) sim['diff'] = model.eval_formula("delta*k-i", sim) model.eval_formula("y(1) - y", sim) sim['ddiff'] = model.eval_formula("diff(1)-diff(-1)", sim)
def test_eval_formula(): from dolo.compiler.eval_formula import eval_formula from dolo import yaml_import, time_iteration, simulate import dolo.config from dolo import perturb model = yaml_import("examples/models/rbc.yaml") dr = perturb(model) sim = simulate(model, dr) sim = sim.sel(N=0) sim = sim.to_pandas() print(sim.columns) rr = eval_formula("delta*k(0)-i(0)", sim, context=model.calibration) rr = eval_formula("y(1) - y(0)", sim, context=model.calibration) sim["diff"] = model.eval_formula("delta*k(0)-i(0)", sim) model.eval_formula("y(1) - y(0)", sim) sim["ddiff"] = model.eval_formula("diff(1)-diff(-1)", sim)
def test_eval_formula(): from dolo.compiler.eval_formula import eval_formula from dolo import yaml_import, time_iteration, simulate import dolo.config from dolo import perturb model = yaml_import('examples/models/rbc.yaml') dr = perturb(model) sim = simulate(model, dr) sim = sim.sel(N=0) sim = sim.to_pandas() print(sim.columns) rr = eval_formula("delta*k(0)-i(0)", sim, context=model.calibration) rr = eval_formula("y(1) - y(0)", sim, context=model.calibration) sim['diff'] = model.eval_formula("delta*k(0)-i(0)", sim) model.eval_formula("y(1) - y(0)", sim) sim['ddiff'] = model.eval_formula("diff(1)-diff(-1)", sim)