Example #1
0
from utils.datasimulation import DataSimulation
from utils.argparsers.simulationargparser import SimulationArgumentParser
from algorithms.PDGD.pdgd import PDGD
from algorithms.PDGD.deeppdgd import DeepPDGD
from algorithms.DBGD.tddbgd import TD_DBGD
from algorithms.DBGD.pdbgd import P_DBGD
from algorithms.DBGD.tdmgd import TD_MGD
from algorithms.DBGD.pmgd import P_MGD
from algorithms.baselines.pairwise import Pairwise
from algorithms.DBGD.neural.pdbgd import Neural_P_DBGD
from algorithms.DBGD.pdbgd_dsp import P_DBGD_DSP
from algorithms.DBGD.pmgd_dsp import P_MGD_DSP
from algorithms.DBGD.tdNSGD import TD_NSGD
from algorithms.DBGD.tdNSGD_dsp import TD_NSGD_DSP

description = 'Run script for testing framework.'
parser = SimulationArgumentParser(description=description)

rankers = []

# Baselines
ranker_params = {'learning_rate_decay': 0.99}

sim_args, other_args = parser.parse_all_args(ranker_params)
run_name = 'attack/TD_DBGD'

rankers.append((run_name, TD_DBGD, other_args))

sim = DataSimulation(sim_args)
sim.run(rankers)
Example #2
0
            args.ind,
            args.seed,
        )

    run_name = dir_name + ranker_name
    ranker = [(run_name, PairRank, other_args)]
    sim = DataSimulation(sim_args)
    sim.run(ranker)


def set_sim_and_run(args):
    cm = args.click_models[0]
    n_impr = args.n_impressions
    n_results = args.n_results
    algo = args.algo.upper()
    dir_name = "algo/{}/{}/{}/{}/".format(algo, cm, n_impr, n_results)

    switcher = {
        "PAIRRANK": lambda: func_pairrank(args, dir_name),
    }

    return switcher.get(algo, lambda: "ERROR: algorithm type not valid")()


# get input parameters
if __name__ == "__main__":
    DESCRIPTION = "Run script for testing framework."
    parser = SimulationArgumentParser(description=DESCRIPTION)
    input_args = parser.parse_sim_args()
    set_sim_and_run(input_args)