R/termPredictors.default.R

In hturner/gnm: Generalized Nonlinear Models

#  Copyright (C) 2005 Heather Turner
#
#  This program is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation; either version 2 or 3 of the License
#  (at your option).
#
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#  A copy of the GNU General Public License is available at
#  http://www.r-project.org/Licenses/

termPredictors.default <- function(object, ...) {
    if (is.null(object$termPredictors)){
        X <- model.matrix(object)
        termPredictors <- t(rowsum(t(X %*% diag(naToZero(coef(object)))),
                                    attr(X, "assign")))
        colnames(termPredictors) <- c("(Intercept)"[0 %in% attr(X, "assign")],
                                       attr(object$terms, "term.labels"))
        termPredictors
    }
    else object$termPredictors
}