inst/pima_Learner.R

library(mlr3)
library(vetiver)
library(plumber)

task = tsk("pima")
learner = lrn("classif.rpart")

learner$train(task)

v <- vetiver_model(learner, "pima_rpart")
v

pr() %>%
  vetiver_api(v) %>%
  pr_run(port = 8088)

library(mlr3)
library(plumber)
library(vetiver)
library(tibble)

task = tsk("pima")
newdata = as_tibble(task$data())[, task$feature_names]

endpoint <- vetiver_endpoint("http://127.0.0.1:8088/predict")
predict(endpoint, newdata)
tidymodels/vetiver documentation built on July 22, 2024, 10:01 p.m.