Python PytorchDPSynthesizer Examples

Example #1

class TestPytorchDPSynthesizer_DPGAN:
    def setup(self):
        self.dpgan = PytorchDPSynthesizer(DPGAN(), GeneralTransformer())

    def test_fit(self):
        self.dpgan.fit(df)
        assert self.dpgan.gan.generator

    def test_sample(self):
        self.dpgan.fit(df)
        sample_size = len(df)
        synth_data = self.dpgan.sample(sample_size)
        assert synth_data.shape == df.shape

Example #2

class TestPytorchDPSynthesizer_PATECTGAN:
    def setup(self):
        self.patectgan = PytorchDPSynthesizer(PATECTGAN(), None)

    def test_fit(self):
        self.patectgan.fit(
            df, categorical_columns=['sex', 'educ', 'race', 'married'])
        assert self.patectgan.gan.generator

    def test_sample(self):
        self.patectgan.fit(
            df, categorical_columns=['sex', 'educ', 'race', 'married'])
        sample_size = len(df)
        synth_data = self.patectgan.sample(sample_size)
        assert synth_data.shape == df.shape

Example #3

 def setup(self):
     self.patectgan = PytorchDPSynthesizer(
         PATECTGAN(regularization='dragan'), None)

Example #4

 def setup(self):
     self.patectgan = PytorchDPSynthesizer(PATECTGAN(), None)

Example #5

 def setup(self):
     self.dpctgan = PytorchDPSynthesizer(DPCTGAN(), None)

Example #6

 def setup(self):
     self.dpgan = PytorchDPSynthesizer(DPGAN(), GeneralTransformer())

Example #7

 def setup(self):
     self.patectgan = PytorchDPSynthesizer(
         PATECTGAN(loss='wasserstein', regularization='dragan'), None)

Example #8

 def setup(self):
     self.pategan = PytorchDPSynthesizer(PATEGAN(), GeneralTransformer())

Example #9

File: test_quail.py Project: soumyarout/smartnoise-sdk

 def QuailSynth(epsilon):
     return PytorchDPSynthesizer(preprocessor=None,
                     gan=PATECTGAN(loss='cross_entropy', batch_size=50, pack=1, sigma=5.0))