Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
# -*- coding: utf-8 -*-
"""Main module."""
import pickle
from copulas.univariate import GaussianUnivariate
from sdv.metadata import Metadata
from sdv.modeler import Modeler
from sdv.models.copulas import GaussianCopula
from sdv.sampler import Sampler
DEFAULT_MODEL = GaussianCopula
DEFAULT_MODEL_KWARGS = {
'distribution': GaussianUnivariate
}
class NotFittedError(Exception):
pass
class SDV:
"""Automated generative modeling and sampling tool.
Allows the users to generate synthetic data after creating generative models for their data.
Args:
model (type):
Class of the ``copula`` to use. Defaults to
``sdv.models.copulas.GaussianCopula``.