Merge pull request #143 from EmilHvitfeldt/modelspec_predict

PR regarding #139

Author: topepo

NAMESPACE

@@ -14,6 +14,7 @@ S3method(predict,"_elnet")
 S3method(predict,"_lognet")
 S3method(predict,"_multnet")
 S3method(predict,model_fit)
+S3method(predict,model_spec)
 S3method(predict,nullmodel)
 S3method(predict_class,"_lognet")
 S3method(predict_class,model_fit)

R/predict.R

@@ -93,7 +93,7 @@
 predict.model_fit <- function (object, new_data, type = NULL, opts = list(), ...) {
   if (any(names(enquos(...)) == "newdata"))
     stop("Did you mean to use `new_data` instead of `newdata`?", call. = FALSE)
-  
+
   type <- check_pred_type(object, type)
   if (type != "raw" && length(opts) > 0)
     warning("`opts` is only used with `type = 'raw'` and was ignored.")
@@ -224,4 +224,7 @@ multi_predict.default <- function(object, ...)
   stop ("No `multi_predict` method exists for objects with classes ",
         paste0("'", class(), "'", collapse = ", "), call. = FALSE)
 
-
+#' @export
+predict.model_spec <- function(object, ...) {
+  stop("You must use `fit()` on your model specification before you can use `predict()`.", call. = FALSE)
+}