X-Git-Url: https://code.communitydata.science/ml_measurement_error_public.git/blobdiff_plain/d0c5766bdf867a81a2477d2cac1d40812110af90..HEAD:/simulations/plot_dv_example.R

diff --git a/simulations/plot_dv_example.R b/simulations/plot_dv_example.R
index 71963b1..88d88a8 100644
--- a/simulations/plot_dv_example.R
+++ b/simulations/plot_dv_example.R
@@ -6,7 +6,7 @@ library(filelock)
 library(argparser)
 
 parser <- arg_parser("Simulate data and fit corrected models.")
-parser <- add_argument(parser, "--infile", default="", help="name of the file to read.")
+parser <- add_argument(parser, "--infile", default="robustness_3_dv.feather", help="name of the file to read.")
 parser <- add_argument(parser, "--remember-file", default="remembr.RDS", help="name of the remember file.")
 parser <- add_argument(parser, "--name", default="", help="The name to safe the data to in the remember file.")
 args <- parse_args(parser)
@@ -86,8 +86,6 @@ build_plot_dataset <- function(df){
 
 change.remember.file(args$remember_file, clear=TRUE)
 sims.df <- read_feather(args$infile)
-sims.df[,Bzx:=NA]
-sims.df[,accuracy_imbalance_difference:=NA]
 plot.df <- build_plot_dataset(sims.df)
 
 remember(plot.df,args$name)
@@ -97,6 +95,7 @@ set.remember.prefix(gsub("plot.df.","",args$name))
 remember(median(sims.df$cor.xz),'med.cor.xz')
 remember(median(sims.df$accuracy),'med.accuracy')
 remember(median(sims.df$error.cor.x),'med.error.cor.x')
+remember(median(sims.df$error.cor.z),'med.error.cor.z')
 remember(median(sims.df$lik.ratio),'med.lik.ratio')