Commit ba7eacbe authored by Tiago Peixoto's avatar Tiago Peixoto
Browse files

blockmodel.py: Better initialization of q_cache

parent e63919e7
...@@ -396,7 +396,7 @@ class BlockState(object): ...@@ -396,7 +396,7 @@ class BlockState(object):
self._state = libinference.make_block_state(self, _get_rng()) self._state = libinference.make_block_state(self, _get_rng())
if deg_corr: if deg_corr:
init_q_cache(max(self.get_E(), self.get_N()) + 1) init_q_cache(max(2 * max(self.get_E(), self.get_N()), 100))
self._entropy_args = dict(adjacency=True, dl=True, partition_dl=True, self._entropy_args = dict(adjacency=True, dl=True, partition_dl=True,
degree_dl=True, degree_dl_kind="distributed", degree_dl=True, degree_dl_kind="distributed",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment