From 131599a3d5b931841d0e62323a5cb9f0d9bba013 Mon Sep 17 00:00:00 2001 From: prise6 Date: Sun, 14 Jun 2015 17:36:35 +0200 Subject: [PATCH] Fix issue trt/outcome --- R/formatRCTDataset.R | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/R/formatRCTDataset.R b/R/formatRCTDataset.R index e78e1a6..a124bbf 100644 --- a/R/formatRCTDataset.R +++ b/R/formatRCTDataset.R @@ -47,12 +47,14 @@ formatRCTDataset <- function(dataset, outcome.field, treatment.field, interactio outcome <- d[, outcome.field.which] if(!is.factor(outcome)) outcome <- as.factor(outcome) if(!length(levels(outcome)) == 2L) stop(sprintf("outcome %s must be binary", outcome.field)) - cat(sprintf("\"%s\" will be the favorable outcome \n", levels(outcome)[2])) + cat(sprintf("\"%s\" will be the favorable outcome \n", levels(outcome)[2])) + d[, outcome.field.which] <- outcome treatment <- d[, treatment.field.which] if(!is.numeric(treatment) & !is.integer(treatment)) treatment <- as.numeric(treatment) if(!( length(unique(treatment)) == 2L & all(c(0,1) %in% unique(treatment)) )) stop(sprintf("%s, response must be numeric:\n 0 = no treatment \n 1 = treatment \n", treatment.field)) + d[, treatment.field.wich] <- treatment predictors <- colnames(dataset)[-c(outcome.field.which, treatment.field.which)]