DataGenerator/NewGoalDirectedTraj.py [27:56]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
size = 9
scale = 5
policy_map = np.zeros((size,size),dtype=int)

# Row wise assignment: 
policy_map[0,:] = 2

policy_map[1,:7] = 2
policy_map[1,7:] = 1

policy_map[2:4,0] = 2
policy_map[2:4,1:4] = 3
policy_map[2:4,4:7] = 2
policy_map[2:4,7:] = 1

policy_map[4,:4] = 3
policy_map[4,4] = 3
policy_map[4,5:] = 1

policy_map[5,:3] = 3
policy_map[5,3:5] = 0
policy_map[5,5:] = 1

policy_map[6,:2] = 3
policy_map[6,2:7] = 0
policy_map[6,7:] = 1

policy_map[7:,0] = 3
policy_map[7:,1:7] = 0
policy_map[7:,7:] = 1
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



DataGenerator/PolicyVisualizer.py [28:57]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
size = 9
scale = 5
policy_map = np.zeros((size,size),dtype=int)

# Row wise assignment: 
policy_map[0,:] = 2

policy_map[1,:7] = 2
policy_map[1,7:] = 1

policy_map[2:4,0] = 2
policy_map[2:4,1:4] = 3
policy_map[2:4,4:7] = 2
policy_map[2:4,7:] = 1

policy_map[4,:4] = 3
policy_map[4,4] = 3
policy_map[4,5:] = 1

policy_map[5,:3] = 3
policy_map[5,3:5] = 0
policy_map[5,5:] = 1

policy_map[6,:2] = 3
policy_map[6,2:7] = 0
policy_map[6,7:] = 1

policy_map[7:,0] = 3
policy_map[7:,1:7] = 0
policy_map[7:,7:] = 1
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -