Commit 37e1083e authored by Tiago Peixoto's avatar Tiago Peixoto
Browse files

Minor fixes to inference tests

parent 8072a1fc
......@@ -194,7 +194,7 @@ for pvals in iter_ranges(pranges):
print("\t mcmc (single flip)", file=out)
state = gen_nested_state(directed, deg_corr, layered, overlap, rec_, rec, dense_bg)
eargs = dict(dl=dl, degree_dl_kind=degree_dl_kind, beta_dl=0.95)
eargs = dict(degree_dl_kind=degree_dl_kind, beta_dl=0.95)
for i in range(5):
print("\t\t", state.mcmc_sweep(beta=0, d=0.5, entropy_args=eargs),
......
......@@ -240,7 +240,7 @@ def get_hist(params, pos=0, check_only=False):
if model == "nsbm":
bs = [g.get_vertices()] if init == "N" else [zeros(1)]
state = state.copy(bs=bs + [zeros(1)] * 6, sampling=True)
state = state.copy(bs=bs + [zeros(1)] * 6)
else:
b = g.get_vertices() if init == "N" else zeros(g.num_vertices())
state = state.copy(b=b)
......
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