grmpy is a Python package for the simulation and estimation of the generalized Roy model.

grmpy

grmpy is an open-source Python package for the simulation and estimation of the generalized Roy model. It serves as a teaching tool to promote the conceptual framework of the generalized Roy model, illustrate a variety of issues in the econometrics of policy evaluation, and showcases basic software engineering practices.
Marginal Treatment Effects (MTE) can be estimated based on a parametric normal model or, alternatively, via the semiparametric method of Local Instrumental Variables (LIV).

You can install grmpy either via pip

$ pip install grmpy

Or download it directly from our GitHub repository and install the package in editable mode

$ git clone https://github.com/OpenSourceEconomics/grmpy.git
$ pip install -e .

Quick Start

Initialization File

grmpy relies on an "initialization.yml" file (referred to as ìnit_file below) to perform both simulation and estimation. For example, check out these two init_files for simulation and parametric estimation as well as a semiparametric estimation setup.

Below you'll find some example code you can copy to jump-start your project.

Simulation

import grmpy

# Specify the initilaization file you want to use, e.g.:
init_file = "ProjectFiles/simulation.yml"

data = grmpy.simulate(init_file)

Estimation

import grmpy

# Specify the initilaization file you want to use, e.g.:
init_file = "ProjectFiles/estimation.yml"

# Parametric Normal Model
rslt = grmpy.fit(init_file, semipar=False)
grmpy.plot_mte(rslt, init_file, color="blue", semipar=False, save_plot="MTE_par.png")

# Local Instrumental Variables (Semiparametric Model)
rslt = grmpy.fit(init_file, semipar=True)
grmpy.plot_mte(rslt, init_file, color="orange", semipar=True, nboot= 250, save_plot="MTE_semipar.png")

Please visit our online documentation for tutorials and more.