[Analogue-commits] r295 - pkg/R

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

Author: gsimpson
Date: 2013-01-03 13:18:29 +0100 (Thu, 03 Jan 2013)
New Revision: 295

Modified:
   pkg/R/crossval.pcr.R
Log:
adds a class and computes LOO estimates as rowMeans

Modified: pkg/R/crossval.pcr.R
===================================================================
--- pkg/R/crossval.pcr.R	2013-01-03 12:07:56 UTC (rev 294)
+++ pkg/R/crossval.pcr.R	2013-01-03 12:18:29 UTC (rev 295)
@@ -54,6 +54,7 @@
                 pred[i, j] <- Xi %*% FIT$B[, j, drop = FALSE] + B0
             }
         }
+        pred <- rowMeans(pred, na.rm = TRUE)
     }
     if(identical(method, "kfold")) {
         ## form ncomp, as k-fold we have ceiling(N / nfold) fewer sites
@@ -157,5 +158,6 @@
         }
         pred <- rowMeans(pred, na.rm = TRUE, dims = 2)
     }
+    class(pred) <- "crossval"
     pred
 }