hsbm_tm.py 文件源码

def inference(self,g):
        """We load a graph-tool graph-object and fit an hsbm:
           - hierarchical
           - nonoverlapping
           - degree-corrected
           We get a state-object which is a 'NestedBlockState'.
           We save as 'state.pkl'
        """
        if self.args.state == None:
            state=gt.minimize_nested_blockmodel_dl(g,deg_corr=True,overlap=False)
            ## save state
            with open(os.path.join(self.out_path,'state.pkl'),'wb') as f:
                pickle.dump(state,f)
        ## if the state already exists, we just load
        else:
            with open(self.args.state,'rb') as f:
                state = pickle.load(f)
        return state