dot-randomize_internal: Add a Random Treatment Indicator Column to a Data Frame

In imt: Impact Measurement Toolkit

Add a Random Treatment Indicator Column to a Data Frame

Description

This function takes a data frame and adds a new column named "treated" with randomly assigned TRUE/FALSE values. Randomization can be done either on the entire data frame or stratified by specified columns. The probability of being assigned to the treatment group can be specified, with a default of 0.5.

Usage

.randomize_internal(data, group_by = NULL, seed = NULL, pr_treated = 0.5)

Arguments

data	The input data frame.
group_by	(Optional) A character vector of column names to stratify the randomization. If provided, the randomization will be done within d each groupefined by the specified columns.
seed	(Optional) An integer to set the random seed for reproducibility.
pr_treated	(Optional) The probability of a row being assigned to the treatment group (TRUE). Default is 0.5.

Value

A new data frame with the added "treated" column.

imt documentation built on Sept. 11, 2024, 5:41 p.m.