Use the discriminator of the model after training

[matteoguarrera]

Problem Description

It seems that the library does not have the possibility to use the discriminator after the training. Is it true?

Expected behavior

I believe that it is possible to add a score function like score_samples(X) that enables the user to see the GAN discriminator output for a given sample.

[fealho]

Hi @matteoguarrera, you are correct, we currently do not plan to give access to the discriminator after training.

[matteoguarrera]

Thanks @fealho for the answer. I can do that and request a pull if you are interested.

[fealho]

I'm glad to hear you would be willing to contribute to the project! Unfortunately, I don't believe this feature is a priority at the moment. Maybe in the future, if more people show interest, we may implement it in SDV, but for now you don't need to worry about it.

[gagank1]

I think this would be useful too. It could be used for checkpointing and early stopping on a validation set held out of the real dataset.

[neuralnets-creator]

Would also be interested in using the discriminator afterwards. Is there any chance to save the trained discriminator to rerun it on unseen data?

[Deepam-Rai]

Same here, even if just to get the model structure of discriminator for documentation purposes.

[Sylv-Lej]

Hi, I have a great interest on this feature also !