0

I am trying to replicate a section of code from the graph-tool cookbook to find the marginal probablity of the number of groups in a graph when using hierarchical partitioning. I however get an error telling me that 'NestedBlockState' object has no attribute 'get_nonempty_B' so presumably I have made a mistake somewhere. Does anybody know where I went wrong?

import graph_tool.all as gt
import cPickle as pickle

g = gt.load_graph('graph_no_multi_reac_type.gt')
gt.remove_parallel_edges(g)

state = gt.minimize_nested_blockmodel_dl(g, deg_corr=True)

state = state.copy(sampling=True)
with open('state_mcmc.pkl','wb') as state_pkl:
    pickle.dump(state,state_pkl,-1)

print 'equilibrating Markov chain'
gt.mcmc_equilibrate(state, wait=1000, mcmc_args=dict(niter=10))

h = np.zeros(g.num_vertices() + 1)

def collect_num_groups(s):
    B = s.get_nonempty_B()
    h[B] += 1

print 'colleting marginals'
gt.mcmc_equilibrate(state, force_niter=10000, mcmc_args=dict(niter=10),
                callback=collect_num_groups)
with open('state_ncnc.pkl','wb') as state_pkl:
    pickle.dump(state,state_pkl,-1)

with open('hist.pkl','wb') as h_pkl:
    pickle.dump(h,h_pkl,-1)

The error I get looks as follows:

Traceback (most recent call last):
  File "num_groups_marg_prob.py", line 42, in <module>
        gt.mcmc_equilibrate(state, force_niter=10000, mcmc_args=dict(niter=10),
  File "/usr/lib/python2.7/dist-packages/graph_tool/inference/mcmc.py", line 172, in mcmc_equilibrate
        extra = callback(state)
  File "num_groups_marg_prob.py", line 35, in collect_num_groups
        def collect_num_groups(s):
AttributeError: 'NestedBlockState' object has no attribute 'get_nonempty_B' 
P-M
  • 1,279
  • 2
  • 21
  • 35

1 Answers1

0

Quoting from an answer from the graph-tool mailing list:

"The error message is clear. This attribute belongs to BlockState, not NestedBlockState. What you wish to do is:

s.levels[0].get_nonempty_B()

"

http://main-discussion-list-for-the-graph-tool-project.982480.n3.nabble.com/self-state-couple-state-state-state-entropy-args-Python-argument-types-did-not-match-C-signature-td4026975.html

P-M
  • 1,279
  • 2
  • 21
  • 35