run_case = 1 ###### Full graph - 2K with triangles + Improved MCMC if run_case == 1: myest = Estimation() myest.load_graph(fname) myest.calcfull_CCK() myest.calcfull_JDD() mygen = Generation() mygen.set_JDD( myest.get_JDD('full') ) mygen.set_KTRI( myest.get_KTRI('full') ) mygen.construct_triangles_2K() mygen.mcmc_improved_2_5_K(nmae_threshold=0.03) mygen.save_graphs('%s_2KT+ImpMCMC_Full' % fname) ####################################################### ###### Full graph - 2K simple + MCMC elif run_case == 2: myest = Estimation() myest.load_graph(fname) myest.calcfull_CCK() myest.calcfull_JDD() mygen = Generation() mygen.set_JDD( myest.get_JDD('full') ) mygen.set_KTRI( myest.get_KTRI('full') ) mygen.construct_simple_2K() mygen.mcmc_random_2_5_K(nmae_threshold=0.03) mygen.save_graphs('%s_2Ksimple+MCMC_Full' % fname)
###### Full graph - 2K with triangles + Improved MCMC if run_case == 1: myest = Estimation() myest.load_graph(fname) myest.calcfull_CCK() myest.calcfull_JDD() myest.estimation_summary() mygen = Generation() mygen.set_JDD(myest.get_JDD('full')) mygen.set_KTRI(myest.get_KTRI('full')) mygen.construct_triangles_2K() mygen.mcmc_improved_2_5_K(error_threshold=error_threshold) mygen.save_graphs('%s_2KT+ImpMCMC_Full' % fname) ####################################################### ###### Full graph - 2K simple + MCMC elif run_case == 2: myest = Estimation() myest.load_graph(fname) myest.calcfull_CCK() myest.calcfull_JDD() myest.estimation_summary() mygen = Generation() mygen.set_JDD(myest.get_JDD('full')) mygen.set_KTRI(myest.get_KTRI('full')) mygen.construct_simple_2K()