Skip to main content

PropensityScoreMatch is a class for matching propensity score and treatment effect

Project description

# CAUSAL INFERENCE AAGM

# Background Propensity score matching is a statistical technique used to estimate the effect of a treatment or intervention on an outcome of interest. It is commonly used in observational studies, where the assignment of treatment or exposure to a particular group is not randomized.

The idea behind propensity score matching is to balance the characteristics of the treatment and control groups by matching individuals with similar propensity scores, which are the probabilities of receiving the treatment or intervention based on observed covariates. This helps to control for confounding factors and reduce selection bias, allowing for a more accurate estimation of the treatment effect.

Overall, propensity score matching is a useful tool for researchers to make causal inferences in observational studies, although it is important to consider the limitations and assumptions of this method.

# Requirements Library This python requires related package more importantly python_requires=’>=3.1’, so that package can be install Make sure the other packages meet the requirements below - pandas>=1.1.5, - numpy>=1.18.5, - scipy>=1.2.0, - matplotlib>=3.1.0, - statsmodels>=0.8.0

# Usage Guide This is a Python class named PropensityScoreMatch. It is designed to perform propensity score matching, a technique used to balance the distribution of confounding variables between treatment and control groups in observational studies. The class has four input arguments:

  • df: a pandas DataFrame containing the data to be analyzed.

  • features: a list of column names in df that contain the variables used to calculate propensity scores.

  • treatment: a string that specifies the name of the column in df that contains the treatment variable.

  • outcome: a string that specifies the name of the column in df that contains the outcome variable.

The output of the class is two pandas DataFrames:

  • df_matched: a DataFrame containing the data for the matched pairs of treated and control observations.

  • df_TE: a DataFrame containing the treatment effect estimates for each variable in features.

In addition to these output DataFrames, the class provides two methods for visualizing the results of the analysis:

  • plot_smd(): a method that generates a plot of standardized mean differences (SMDs) between the treatment and control groups for each variable in features.

  • plot_individual_treatment(): a method that generates a plot of the individual treatment effects for each observation in df_matched.

For Analysis: - plot_smd() : plotting the df_smd

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

causal_inference_aagm-0.0.3.tar.gz (2.4 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page