DataGenerator/DirectedContinuousNonZero.py [10:30]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
number_datapoints = 50000
number_timesteps = 25

x_array_dataset = np.zeros((number_datapoints, number_timesteps, 2))
a_array_dataset = np.zeros((number_datapoints, number_timesteps-1, 2))
y_array_dataset = np.zeros((number_datapoints, number_timesteps-1),dtype=int)
b_array_dataset = np.zeros((number_datapoints, number_timesteps-1),dtype=int)
goal_array_dataset = np.zeros((number_datapoints, 1),dtype=int)

action_map = np.array([[0,-1],[-1,0],[0,1],[1,0]])
start_states = np.array([[-2,-2],[-2,2],[2,-2],[2,2]])*5
valid_options = np.array([[2,3],[3,0],[1,2],[0,1]])

for i in range(number_datapoints):

	if i%1000==0:
		print("Processing Datapoint: ",i)
	b_array_dataset[i,0] = 1.

	# Select one of four starting points. (-2,-2), (-2,2), (2,-2), (2,2)
	goal_array_dataset[i] = np.random.random_integers(0,high=3)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



DataGenerator/DirectedContinuousTrajs.py [10:30]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
number_datapoints = 50000
number_timesteps = 25

x_array_dataset = np.zeros((number_datapoints, number_timesteps, 2))
a_array_dataset = np.zeros((number_datapoints, number_timesteps-1, 2))
y_array_dataset = np.zeros((number_datapoints, number_timesteps-1),dtype=int)
b_array_dataset = np.zeros((number_datapoints, number_timesteps-1),dtype=int)
goal_array_dataset = np.zeros((number_datapoints, 1),dtype=int)

action_map = np.array([[0,-1],[-1,0],[0,1],[1,0]])
start_states = np.array([[-2,-2],[-2,2],[2,-2],[2,2]])*5
valid_options = np.array([[2,3],[3,0],[1,2],[0,1]])

for i in range(number_datapoints):

	if i%1000==0:
		print("Processing Datapoint: ",i)
	b_array_dataset[i,0] = 1.

	# Select one of four starting points. (-2,-2), (-2,2), (2,-2), (2,2)
	goal_array_dataset[i] = np.random.random_integers(0,high=3)
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -