Add survfit class designation to h2o.coxph survfit output.

AlicTsai · Jan 23, 2015 · b504526 · b504526
1 parent fa8af63
commit b504526
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/R/h2o-package/R/Classes.R b/R/h2o-package/R/Classes.R
@@ -223,6 +223,7 @@ function(formula, newdata, conf.int = 0.95,
     res$cumhaz <- outer(res$cumhaz, pred, FUN = "*")
   res$std.err <- NULL
   res$surv <- exp(- res$cumhaz)
+  class(res) <- c("survfit.H2OCoxPHModel", "survfit.cox", "survfit")
   return(res)
 
   # Code below assumes calculation in H2O