-
Notifications
You must be signed in to change notification settings - Fork 0
/
ask_xai_questions.py
30 lines (26 loc) · 1.35 KB
/
ask_xai_questions.py
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
import sys
sys.path.append("./Code")
sys.path.append("./Code/PantheonRL/overcookedgym/human_aware_rl/overcooked_ai")
from Code.Explainability.Discretizers.Discretizer10 import Discretizer10
from Code.Explainability.Discretizers.Discretizer11 import Discretizer11
from Code.Explainability.Discretizers.Discretizer12 import Discretizer12
from Code.Explainability.Discretizers.Discretizer13 import Discretizer13
from Code.Explainability.Discretizers.Discretizer14 import Discretizer14
from Code.Explainability.PartialPolicyGraph import PartialPolicyGraph
from Code.Explainability.CompletePolicyGraph import CompletePolicyGraph
from Code.Experiment import Experiment
if __name__ == '__main__':
# agent_ids = ['_simple1M', '_unident_s1M', '_random0_1M', '_random1_1M', '_random3_1M']
# layouts = ['simple', 'unident_s', 'random0', 'random1', 'random3']
# Run one experiment only
experiment = Experiment(id='2_Experiment_500_25', rl_agent_id='_simple1M',
discretizer=Discretizer11,
pg_algorithm=CompletePolicyGraph,
layout='simple',
description="")
experiment.run(train=False,
test=False,
ask_questions_xai=True,
subgraph=None)
# subgraph=5 -> Partial
# subgraph=2,3 -> Complete