FairCompromises/main.py at main · TDI-Lab/FairCompromises · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
import  params_greedy as pms
from run import experiment_part
import ipdb
from utils import *
import random
import cProfile
from voting_agent.BQ import BranchingQN
import copy

BAM="greedy"

if __name__ == "__main__":
    path = "aarau_data/switzerland_aarau_2023.pb"
    instance, profile, voter_preferences, voter_ballot = get_voter_data(path)
    device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

    params = copy.deepcopy(pms.PARAMS)
    params["experiment_description"] = ""
    params['data path']=  path
    params['samples'] = 10
    params['robustness']=1
    params['experiment_name']= "aarau_greedy"

    params['experiments'][BAM]['training_config']['num_iters']=300
    params['experiments'][BAM]['training_config']['algorithm']=BranchingQN
    params['experiments'][BAM]['env_config']['ballot']='cumulative'
    params['experiments'][BAM]['env_config']['project_voting_cap']=False
    params['experiments'][BAM]['env_config']['voted_for']=True
    params['experiments'][BAM]['env_config']['utilitarian_altruism'] = False
    params['experiments'][BAM]['env_config']['egalitarian_altruism'] = False

    TESTING=True

    if TESTING:
        instance, profile, voter_preferencesF, voter_ballotF = get_voter_data(path)
        params['experiments'][BAM]['training_config']['num_iters']=100
        params['samples'] = 5
        params['num_agents'] = 30
        agent_sample = random.sample(sorted(voter_preferencesF), params['num_agents'])
        voter_preferences = {voter:voter_preferencesF[voter] for voter in agent_sample}
        voter_ballot = {voter:voter_ballotF[voter] for voter in agent_sample}

    for exp_name, exp_params in params['experiments'].items():
        params['experiments'][exp_name]['env_config']['num_agents']=len(voter_preferences)
        for sample in range(params['samples']):
            for time in range(params['robustness']):
                exp_params['exp_part']=exp_name+str(len(voter_preferences))+"Agents"+"Sample"+str(sample)+"Time"+str(time)
                experiment_part(
                    exp_params, params['experiment_name'],
                    voter_preferences=voter_preferences,
                    instance=instance,
                    profile=profile,
                    voter_ballot=voter_ballot,
                    device=device)
                #cProfile.run("experiment_part(exp_params, params.PARAMS['experiment_name'],voter_preferences=voter_preferences[time],aarau_instance=aarau_instance,aarau_profile=aarau_profile, voter_ballot)")