search

pymc-devs / pymc

Bayesian Modeling and Probabilistic Programming in Python
https://docs.pymc.io/
Other
8.71k stars 2.01k forks source link

Feature request: Option to ignore imputation in the `pm.model_to_graphviz` plate model #4875

Open michaelosthege opened 3 years ago

michaelosthege commented 3 years ago

Description of your problem

Using the imputation feature leads to an automatic split-up of the model variables. In the pm.model_to_graphviz visualization that causes arrows to go in the reverse direction:

grafik

Versions and main components

ricardoV94 commented 3 years ago

You can exclude model.auto_deterministics from the model_graphviz output, and I think it would get rid of the upper one

michaelosthege commented 3 years ago

But I want to get rid of the automatically created imputation variables, not the meaningful ones!

ricardoV94 commented 3 years ago

I see, that won't be as simple. That's why I mentioned the other option :D