How to render a semi supervised model?

Hi,
I am trying to plot a model where I would like to also do automatized rendering, however I can only set obs=data to denote that sample is supervised, and if I keep obs empty, then the node is unsupervised.
However, I would like to denote a semi supervised setting where we would ‘paint’ the half of the sample, like in SS-VAE figures in Pyro website, as below:

Can you let me know how to make that happen?
Thanks!

This is a cool idea, you should open an feature request issue. I think we could make this work by passing two sets of *args,kwargs**.

Actually, this was simple to implement, here’s a PR on github.
image

1 Like

Thanks!