fix a bug in predict.ontologyLoss

pradosj · Feb 13, 2018 · d406623 · d406623
1 parent f7df11a
commit d406623
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bmrm/R/loss_convex.R b/bmrm/R/loss_convex.R
@@ -427,8 +427,8 @@ ontologyLoss <- function(x,y,l=1 - table(seq_along(y),y),dag=diag(nlevels(y))) {
 #' @export
 predict.ontologyLoss <- function(object,x,...) {
   W <- array(object,attr(object,"model.dim"))
-  W <- tcrossprod(W,attr(object,"model.dag"))
   dimnames(W) <- attr(object,"model.dimnames")
+  W <- tcrossprod(W,attr(object,"model.dag"))
   f <- x %*% W
   y <- max.col(f,ties.method="first")
   y <- factor(colnames(W)[y],colnames(W))