Description Usage Arguments Examples
This functions serializes a Spark pipeline model into an MLeap bundle.
1 | ml_write_bundle(x, sample_input, path, overwrite = FALSE)
|
x |
A Spark pipeline model object. |
sample_input |
A sample input Spark DataFrame with the expected schema. |
path |
Where to save the bundle. |
overwrite |
Whether to overwrite an existing file, defaults to |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | ## Not run:
library(sparklyr)
sc <- spark_connect(master = "local")
mtcars_tbl <- sdf_copy_to(sc, mtcars, overwrite = TRUE)
pipeline <- ml_pipeline(sc) %>%
ft_binarizer("hp", "big_hp", threshold = 100) %>%
ft_vector_assembler(c("big_hp", "wt", "qsec"), "features") %>%
ml_gbt_regressor(label_col = "mpg")
pipeline_model <- ml_fit(pipeline, mtcars_tbl)
model_path <- file.path(tempdir(), "mtcars_model.zip")
ml_write_bundle(pipeline_model,
mtcars_tbl,
model_path,
overwrite = TRUE)
## End(Not run)
|
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.