Add tidy method for Liblinear parsnip models

Author: juliasilge

NAMESPACE

@@ -66,6 +66,7 @@ S3method(req_pkgs,model_fit)
 S3method(req_pkgs,model_spec)
 S3method(required_pkgs,model_fit)
 S3method(required_pkgs,model_spec)
+S3method(tidy,"_LiblineaR")
 S3method(tidy,"_elnet")
 S3method(tidy,"_fishnet")
 S3method(tidy,"_lognet")

R/tidy_liblinear.R

@@ -0,0 +1,17 @@
+#' tidy methods for LiblineaR models
+#'
+#' `tidy()` methods for the various `LiblineaR` models that return the
+#' coefficients from the `parsnip` model fit.
+#' @param x A fitted `parsnip` model that used the `LiblineaR` engine.
+#' @param ... Not used
+#' @return A tibble with columns `term` and `estimate`.
+#' @keywords internal
+#' @export
+
+tidy._LiblineaR <- function(x, ...) {
+  check_installs(x$spec)
+  ret <- tibble(colnames(x$fit$W), x$fit$W[1,])
+  colnames(ret) <- c("term", "estimate")
+
+  ret
+}