Conditional sampling from DDPM about synthcity

[TZTsai]

@dionman
Which version are you using? Could you try the latest version in my fork? RuntimeError: Conditional mismatch. Got inference conditional, without any training conditional is probably due to a previous bug that the Plugin requires an explicit cond argument during fit if you want to generate with a cond as well, but DDPM automatically sets the labels as the cond if is_classification=True. This has already been fixed.

For the second problem, yes the cond should have a float or int data type for the moment, because otherwise it needs one-hot encoding, while in the diffusion model it is assumed to be 1-dimensional. Perhaps you can also use LabelEncoder to encode the cond if it has an object dtype and provide it to the fit method as follows.

[dionman]

I'm using release v0.2.5. Is it fixed at this version of the code?

[TZTsai]

I'm using release v0.2.5. Is it fixed at this version of the code?

I checked that commit. Sorry it is not fixed there.

[dionman]

okay, what workaround should I use at the moment? Import the library from your fork?

[TZTsai]

You can replace Line 214-224 in src/synthcity/plugins/generic/plugin_ddpm.py with the following code:

        df = self.encoder.fit_transform(df)

        if cond is not None:
            if type(cond) is str:
                cond = df[cond]
            cond = pd.Series(cond, index=df.index)
            self.expecting_conditional = True

        # NOTE: cond may also be included in the dataframe
        self.model.fit(df, cond, **kwargs)
        self.loss_history = self.model.loss_history

        return self

[dionman]

do you mean lines 218-224?

[TZTsai]

do you mean lines 218-224?

No, should be from line 214, starting with else:.

[dionman]

https://github.com/TZTsai/synthcity/blob/e336d3cd42516cef27ac4ad60ac0df61bd92f1a6/src/synthcity/plugins/generic/plugin_ddpm.py#L157
self.encoder also needs to be added earlier in this script
...which implies changing a few more things

[TZTsai]

TZTsai/synthcity@e336d3c/src/synthcity/plugins/generic/plugin_ddpm.py#L157 self.encoder also needs to be added earlier in this script ...which implies changing a few more things

Sorry I forgot about that. You can comment that line for the moment. Can it run after that?

[dionman]

Thanks. Is there any cleaner way to enable this after the most recent commits in the main? Can I use any branch in your fork?

[TZTsai]

Sorry I just found my last reply wasn't sent out, probably due to internet reasons. In the latest version of the main branch, this should work fine. Have you checked it?

[dionman]

thanks, yes, this works