From de17a6d755e9369a91abdb06562ee93d7b323bbd Mon Sep 17 00:00:00 2001
From: Gertjan van den Burg <gertjanvandenburg@gmail.com>
Date: Wed, 4 Apr 2018 15:06:33 -0400
Subject: Adhere to generic function signatures

---
 R/fitted.gensvm.R | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

(limited to 'R/fitted.gensvm.R')

diff --git a/R/fitted.gensvm.R b/R/fitted.gensvm.R
index 72b5db6..be97223 100644
--- a/R/fitted.gensvm.R
+++ b/R/fitted.gensvm.R
@@ -3,7 +3,7 @@
 #' @description This function shows the fitted class labels of training data 
 #' using a fitted GenSVM model.
 #'
-#' @param fit Fitted \code{gensvm} object
+#' @param object Fitted \code{gensvm} object
 #' @param \dots further arguments are passed to predict
 #'
 #' @return a vector of class labels, with the same type as the original class 
@@ -22,8 +22,9 @@
 #' \code{\link{plot.gensvm}}, \code{\link{predict.gensvm.grid}}, 
 #' \code{\link{gensvm}}, \code{\link{gensvm-package}}
 #'
+#' @method fitted gensvm
+#'
 #' @export
-#' @aliases fitted
 #'
 #' @examples
 #' x <- iris[, -5]
@@ -36,7 +37,8 @@
 #' # compute the accuracy with gensvm.accuracy
 #' gensvm.accuracy(y, yhat)
 #'
-fitted.gensvm <- function(fit, ...)
+fitted.gensvm <- function(object, ...)
 {
-    return(predict(fit, ...))
+    x <- eval.parent(object$call$x)
+    return(predict(object, x, ...))
 }
-- 
cgit v1.2.3