pyhgf
pyhgf copied to clipboard
PyHGF: A neural network library for predictive coding
The data format we are currently using to represent `attributes` and `edges` is suboptimal regarding JAX transformation. Per JAX standard, PyTrees are only accessible at compile time and cannot be...
The `plot_network` function is currently using GraphViz, however, it would be convenient to have something more Matplotlib-compatible, and [NetworkX](https://networkx.org/) seems to be the go-to library. This would also let us...
Add standard reinforcement learning algorithms and provide some examples of model comparison.
Add a method for pretty printing models' parameters.
This for loop is slowing model fitting when many datasets/models are provided. https://github.com/ilabcode/ghgf/blob/8fec5d71e1677e85103ae9dd79dda26056b5b06f/ghgf/distribution.py#L167 We should use scan or for loop instead, but this might require careful refactoring. It is not...
Creating a graphical interface to edit node structures manually. This should ideally create an object that can be an input to both the Python and Julia HGFs.