コード例 #1
0
    #     jc.JordanCenter(),
    #     ri.ReverseInfection(),
    #     di.DynamicImportance(),
    #     prior_detector8,
    #     gsba.GSBA(prior_detector1),
    #     gsba_bao7.GSBA_coverage_7(prior_detector1),
    #
    # ]

    methods = [
        rc.RumorCenter(),
        dc.DistanceCenter(),
        jc.JordanCenter(),
        ri.ReverseInfection(),
        di.DynamicImportance(), prior_detector8,
        gsba.GSBA(prior_detector1),
        gsba_bao7.GSBA_coverage_7(prior_detector1),
        gsba_bao9.GSBA_coverage_9(prior_detector1)
    ]

    logger = log.Logger(logname='../data/main_5regular_tree1229.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/5regular_tree_2000.txt", weighted=0)
    d.debug = False
    test_num = 10
コード例 #2
0
ファイル: main_ulbaa.py プロジェクト: yuhuangithub/research
from experiment import Experiment

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(),
        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 = [
        enum.Likelihoods(prior_detector1),
        ulbaa.ULBAA(prior_detector1),
        gsba.GSBA(prior_detector1)
    ]
    # methods = [rc.RumorCenter(), dc.DistanceCenter(), jc.JordanCenter(), ri.ReverseInfection(), di.DynamicImportance(),
    #            gsba.GSBA(prior_detector1),gsba.GSBA(prior_detector3),gsba.GSBA(prior_detector4),
    #            ulbaa.ULBAA(prior_detector1), ulbaa.ULBAA(prior_detector3), ulbaa.ULBAA(prior_detector4)]

    logger = log.Logger(logname='../data/main_ulbaa.log',
コード例 #3
0
ファイル: test.py プロジェクト: yuhuangithub/research
import map_gsba3 as gsba3
import map_rsa as rsa

import numpy as np
from experiment import Experiment

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 = [
        gsba.GSBA(prior_detector0),
        gsba.GSBA(prior_detector1),
        gsba.GSBA(prior_detector3),
        gsba.GSBA(prior_detector4),
        gsba3.GSBA(prior_detector0),
        gsba3.GSBA(prior_detector1),
        gsba3.GSBA(prior_detector3),
        gsba3.GSBA(prior_detector4)
    ]

    logger = log.Logger(logname='../data/main_scale_free3.log',
                        loglevel=logging.INFO,
                        logger="experiment").get_log()
    experiment = Experiment(methods, logger)
    experiment.propagation_model = 'SI'
コード例 #4
0
ファイル: test3.py プロジェクト: yuhuangithub/research
            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")
    # d = data.Graph("../data/small-world.ws.v100.e500.gml", weighted=1)
    d = data.Graph("../data/scale-free.ba.v500.e996.gml", weighted=1)
    #d = data.Graph("../data/power-grid.txt")