import rumor_center as rc import dmp2 import map_bfsa_parallel as bfsa_p import prior import numpy as np from experiment import Experiment import map_ulbaa as ulbaa if __name__ == '__main__': prior_detector0 = prior.Uniform() prior_detector1 = rc.RumorCenter() prior_detector2 = dmp2.DynamicMessagePassing() prior_detector3 = dc.DistanceCenter() prior_detector4 = jc.JordanCenter() prior_detector5 = ri.ReverseInfection() methods = [rc.RumorCenter(), dc.DistanceCenter(), jc.JordanCenter(), gsba.GSBA(prior_detector1),gsba.GSBA(prior_detector3),gsba.GSBA(prior_detector4)] # methods = [dc.DistanceCenter()] #methods = [bfsa_p.BFSA(prior_detector1)] # methods = [dmp2.DynamicMessagePassing()] logger = log.Logger(logname='../data/main_power_grid1202.log', loglevel=logging.INFO, logger="experiment").get_log() experiment = Experiment(methods, logger) experiment.propagation_model = 'SI' start_time = clock() print "Starting..." # d = data.Graph("../data/power-grid.txt") d = data.Graph("../data/power-grid.gml", weighted=1)
import map_gsba_bao5 as gsba_bao5 import map_gsba_bao6 as gsba_bao6 import map_gsba_bao8 as gsba_bao8 import map_gsba_bao11 as gsba_bao11 import map_gsba_bao9 as gsba_bao9 import map_gsba_bao10 as gsba_bao10 import map_gsba_bao12 as gsba_bao12 import map_gsba_bao7 as gsba_bao7 if __name__ == '__main__': prior_detector0 = prior.Uniform() prior_detector1 = rc.RumorCenter() prior_detector2 = dmp2.DynamicMessagePassing() prior_detector3 = dc.DistanceCenter() prior_detector4 = jc.JordanCenter() prior_detector5 = ri.ReverseInfection() prior_detector7 = epa.EPA_center() prior_detector8 = epa2.EPA_center_weight() # 有权重版本 prior_detector9 = bc.Belief_coverage_center() # 置信传播算法。 # methods = [rc.RumorCenter(), dc.DistanceCenter(), jc.JordanCenter(),ri.ReverseInfection(),prior_detector2, # gsba.GSBA( prior_detector1),gsba.GSBA(prior_detector2), gsba.GSBA( prior_detector3), # gsba.GSBA(prior_detector4), gsba.GSBA( prior_detector5)] # methods = [rc.RumorCenter(), dc.DistanceCenter(), jc.JordanCenter(), ri.ReverseInfection(), di.DynamicImportance(), # gsba.GSBA(prior_detector1),gsba.GSBA(prior_detector3),gsba.GSBA(prior_detector4),] ''' 它的东西 ''' # methods = [rc.RumorCenter(), dc.DistanceCenter(), jc.JordanCenter(),
if l==0 : continue r = sum(self.precision[test][m.method_name]) / l, sum(self.error[test][m.method_name]) / l, sum( self.topological_error[test][m.method_name]) / l, sum(self.ranking[test][m.method_name]) / l, m.method_name, l print r logger.info(r) if __name__ == '__main__': logger = log.Logger(logname='../data/test3.log', loglevel=logging.INFO, logger="experiment").get_log() experiment = Experiment() experiment.logger = logger prior_detector0 = prior.Uniform() prior_detector1 = rc.RumorCenter() prior_detector2 = dmp2.DynamicMessagePassing() prior_detector3 = dc.DistanceCenter() prior_detector4 = jc.JordanCenter() prior_detector5 = ri.ReverseInfection() methods = [rc.RumorCenter(), dc.DistanceCenter(), jc.JordanCenter(),ri.ReverseInfection(),prior_detector2, gsba.GSBA( prior_detector1),gsba.GSBA(prior_detector2), gsba.GSBA( prior_detector3), gsba.GSBA(prior_detector4), gsba.GSBA( prior_detector5)] methods = [rc.RumorCenter(), dc.DistanceCenter(), jc.JordanCenter(), ri.ReverseInfection(), di.DynamicImportance(), prior_detector2, gsba.GSBA(prior_detector0), gsba.GSBA(prior_detector1), gsba.GSBA( prior_detector3), gsba.GSBA(prior_detector4), gsba.GSBA( prior_detector5), gsba.GSBA(prior_detector2), bfsa_p.BFSA(prior_detector1)] methods = [bfsa.BFSA(prior_detector1)] # methods = [dmp2.DynamicMessagePassing()] experiment.methods = methods start_time = clock() print "Starting..." d = data.Graph("../data/test_category.txt", weighted=1) # d = data.Graph("../data/karate_club.gml")