from collections import OrderedDict import numpy as np import dqn def test_agent(): case = [1,0,1,0,0,0,0,0,0,0,0,0,0,0,0] event = 3 process = [1,4,2,2,2,4,0,2,3,3,20,20,20] state = OrderedDict() state['case'] = np.asarray(case) state['event'] = event state['process'] = np.asarray(process) print(state) print(dqn.deploy(state)) test_agent()