Calling SDV in an R6 class

sdv-r6.R

library(reticulate)

# Ignore this ----

use_python("/opt/homebrew/bin/python3.10")

# reticulate works ----

sdv <- import("sdv")

# Generate some example data
get_available_demos <- sdv$datasets$demo$get_available_demos
x <- get_available_demos(modality = "single_table")

# Detect metadata
SingleTableMetadata <- sdv$metadata$SingleTableMetadata
metadata <- SingleTableMetadata()
metadata$detect_from_dataframe(data = x)

# Create synthesizer instance
GaussianCopulaSynthesizer <- sdv$single_table$copulas$GaussianCopulaSynthesizer
synthesizer <- GaussianCopulaSynthesizer(metadata)

# Fit model and inference
synthesizer$fit(x)
synthesizer$sample(num_rows = 10L)

# R6 works ----

library(R6)

GaussianCopulaSynthesizer <- R6Class(
  "GaussianCopulaSynthesizer",
  list(
    metadata = NULL,
    synthesizer = NULL,
    initialize = function(metadata) {
      self$metadata <- metadata
      self$synthesizer <- sdv$single_table$copulas$GaussianCopulaSynthesizer(metadata)
    },
    fit = function(data) {
      self$synthesizer$fit(data)
    },
    sample = function(num_rows) {
      self$synthesizer$sample(num_rows = as.integer(num_rows))
    }
  )
)

gcs <- GaussianCopulaSynthesizer$new(metadata)
gcs$fit(x)
gcs$sample(10)