[Mboost-commits] r722 - pkg/mboostPatch/R

[noreply at r-forge.r-project.org]

Author: hofner
Date: 2013-07-09 17:13:38 +0200 (Tue, 09 Jul 2013)
New Revision: 722

Modified:
   pkg/mboostPatch/R/plot.R
Log:
minor bugfix in plot.mboost when data contains missing values


Modified: pkg/mboostPatch/R/plot.R
===================================================================
--- pkg/mboostPatch/R/plot.R	2013-06-27 12:30:40 UTC (rev 721)
+++ pkg/mboostPatch/R/plot.R	2013-07-09 15:13:38 UTC (rev 722)
@@ -64,15 +64,15 @@
 
             if (ncol(data) == 1) {
                 if (!add){
-                    plot(sort(data[[1]]), pr[order(data[[1]])], type = type,
+                    plot(sort(data[[1]]), pr[order(data[[1]], na.last = NA)], type = type,
                          xlab = xl, ylab = yl, ylim = ylim, ...)
                     if (rug) rug(data[[1]], col = rugcol)
                 } else {
                     if (is.factor(data[[1]])){
-                        boxplot(pr[order(data[[1]])] ~ sort(data[[1]]),
+                        boxplot(pr[order(data[[1]], na.last = NA)] ~ sort(data[[1]]),
                                 add = TRUE, ...)
                     } else {
-                        lines(sort(data[[1]]), pr[order(data[[1]])], type =
+                        lines(sort(data[[1]]), pr[order(data[[1]], na.last = NA)], type =
                               type, ...)
                         if (rug){
                             rug(data[[1]], col = rugcol)
@@ -106,7 +106,7 @@
                         tmp[[v]] <- vv
                         mean(predict(x, newdata = tmp, which = w))
                     })
-                    plot(sort(data[[v]]), pardep[order(data[[v]])], type = type,
+                    plot(sort(data[[v]]), pardep[order(data[[v]], na.last = NA)], type = type,
                          xlab = v, ylab = "Partial Dependency", ylim = ylim, ...)
                 }
             }