tests/testthat/test-svm-serde.R

In cuda.ml: R Interface for the RAPIDS cuML Suite of Libraries

context("(de)serialization of SVM models")

test_that("SVM regressor can be serialized and unserialized correctly", {
  model <- cuda_ml_svm(formula = mpg ~ ., data = mtcars, kernel = "rbf")
  model_state <- cuda_ml_serialize(model)

  expected_preds <- predict(model, mtcars)
  actual_preds <- predict_in_sub_proc(
    model_state,
    data = mtcars,
    expected_mode = "regression",
    expected_model_cls = "cuda_ml_svr"
  )

  expect_equal(expected_preds, actual_preds)
})