Is it possible to do model fitting using this package?

[yaowang1111]

Hi~

I want to know if this package can fit model to behavior datas collected by my experiment?
Or will there be future plans to implement this functionality?

Thanks~

[conorheins]

Hi @yaowang1111 ,
Thanks for your question -- unfortunately we can't do model-fitting yet, but this is high on the priority list. We are actively working on a new backend for pymdp that would allow us to fit the parameters of active inference POMDP agents using inference techniques from Numpyro, and to take advantage of GPU acceleration as well. I can't give you a timeline on this project at the moment, but I'll leave this issue open and update you with more information, once this functionality becomes available.

[yaowang1111]

Actually I'm woking on an active inference simulation project using pyro. I think Active inference shouldn't be confined to discrete matrix situations. Generalising it to any general probablistic models would benefit the theory. And pyro seems to be a suitable tool.
I have some very basic understanding of pyro now.
Can i somehow participate in this new version, despite of my limited coding skills?

[conorheins]

Hi @yaowang1111, sorry for the late reply on this! That sounds really cool, namely doing active inference in pyro.

It would be great to discuss more the involvement in the numpyro version of the package. At the moment this is a side-project for both myself and the other person (@dimarkov) working on this project, so we haven't had a chance to work on it in a while. Once we've fired that branch back up I'll get back and touch and we can catch you up on what we've done so far

[conorheins]

I'm going to move this to a discussion since it's more relevant there.