import reverse_infection as ri
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")
import rumor_coverage_center as rumor_coverage
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),]
'''
它的东西
'''
l = len(self.precision[test][m.method_name])*1.0
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)
