brendanhasz / probflow

A Python package for building Bayesian models with TensorFlow or PyTorch
http://probflow.readthedocs.io
MIT License
170 stars 17 forks source link

Bayesian Network Implementation? #69

Open vahndi opened 3 years ago

vahndi commented 3 years ago

Hi Brendan, thanks for the library - really nice.

I'm wondering if there's a way to construct a simple Bayes Net with mixed data types using probflow. The model I'm thinking of would look something like:

I realize this looks a lot like a standard Logistic Regression model, but the difference is:

Do you think this is possible by inheriting from pf.CategoricalModel? If so, any pointers on building it?

Cheers,

Vahndi