ryansmcgee / seirsplus

Models of SEIRS epidemic dynamics with extensions, including network-structured populations, testing, contact tracing, and social distancing.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Running seirsplus on top of a graph_tool network.

denis-pickcells opened this issue · comments

I've started working with the seirsplus framework with a networkx created SBM network, but it became very memory-intensive and slow. Is there a way we can use graph_tool networks to work with the seirsplus models?

I have converted the graph_tool to an adjacency matrix and used it to populate the SEIRSNetworkModel. The problem is that, even though it accepts numpy.ndarrays as input for G parameter, this parameter is treated as a graph in:
transitionNode_GNbrs = list(self.G[transitionNode].keys()) transitionNode_GQNbrs = list(self.G_Q[transitionNode].keys())