X-Git-Url: https://code.communitydata.science/ml_measurement_error_public.git/blobdiff_plain/8ac33c14d7e7874bf283aa9c252fa06566dc8b15..d8bc08f18f8c2128369ee959196e0e6080a11689:/simulations/03_depvar.R

diff --git a/simulations/03_depvar.R b/simulations/03_depvar.R
index dde1bee..f0064f2 100644
--- a/simulations/03_depvar.R
+++ b/simulations/03_depvar.R
@@ -76,12 +76,13 @@ parser <- add_argument(parser, "--prediction_accuracy", help='how accurate is th
 parser <- add_argument(parser, "--Bxy", help='coefficient of x on y', default=0.01)
 parser <- add_argument(parser, "--Bzy", help='coeffficient of z on y', default=-0.01)
 parser <- add_argument(parser, "--Bzx", help='coeffficient of z on x', default=-0.5)
+parser <- add_argument(parser, "--B0", help='Base rate of y', default=0.5)
 parser <- add_argument(parser, "--outcome_formula", help='formula for the outcome variable', default="y~x+z")
 parser <- add_argument(parser, "--proxy_formula", help='formula for the proxy variable', default="w_pred~y")
 
 args <- parse_args(parser)
 
-B0 <- 0
+B0 <- args$B0
 Bxy <- args$Bxy
 Bzy <- args$Bzy
 Bzx <- args$Bzx