test_AE Namespace Reference

Classes
class	Tester

Variables
bool	TEST_TRAINED = True
int	seed = 3
string	name = 'Billiard_AE'
	device = torch.device('cpu')
string	load_path_AES = '/home/giuseppe/src/taxons/experiments/{}_Surprise/{}'
string	load_path_AEN = '/home/giuseppe/src/taxons/experiments/{}_Novelty/{}'
string	load_path_Mixed = '/home/giuseppe/src/taxons/experiments/{}_Mixed/{}'
string	load_path_NT = '/home/giuseppe/src/taxons/experiments/{}_NoTrain/{}'
string	env_tag = "Billiard-v0"
int	number_of_samples = 10
	AEN = Tester(load_path_AEN, device)
	AES = Tester(load_path_AES, device)
	Mixed = Tester(load_path_Mixed, device)
	NT = Tester(load_path_NT, device)
list	x_test = []
	env = gym.make(env_tag)
	RANDOM_BALL_INIT_POSE
	RANDOM_ARM_INIT_POSE
	CoM = np.array([env.env.data.qpos[:2]])
	tmp = env.render(mode='rgb_array')
	images_test = torch.Tensor(x_test).permute(0, 3, 1, 2).to(device)
	fig
	ax
	N_loss
	N_f
	N_y = N_y.permute(0, 2, 3, 1)[0]
	S_loss
	S_f
	S_y = S_y.permute(0, 2, 3, 1)[0]
	M_loss
	M_f
	M_y = M_y.permute(0, 2, 3, 1)[0]
	NT_loss
	NT_f
	NT_y = NT_y.permute(0, 2, 3, 1)[0]
	subs = AEN.selector.subsample(images_test[k:k+1])
	img = np.array(N_y.cpu().data * 255)

Variable Documentation

test_AE.AEN = Tester(load_path_AEN, device)

test_AE.AES = Tester(load_path_AES, device)

test_AE.ax

test_AE.CoM = np.array([env.env.data.qpos[:2]])

test_AE.device = torch.device('cpu')

test_AE.env = gym.make(env_tag)

string test_AE.env_tag = "Billiard-v0"

test_AE.fig

test_AE.images_test = torch.Tensor(x_test).permute(0, 3, 1, 2).to(device)

test_AE.img = np.array(N_y.cpu().data * 255)

string test_AE.load_path_AEN = '/home/giuseppe/src/taxons/experiments/{}_Novelty/{}'

string test_AE.load_path_AES = '/home/giuseppe/src/taxons/experiments/{}_Surprise/{}'

string test_AE.load_path_Mixed = '/home/giuseppe/src/taxons/experiments/{}_Mixed/{}'

string test_AE.load_path_NT = '/home/giuseppe/src/taxons/experiments/{}_NoTrain/{}'

test_AE.M_f

test_AE.M_loss

test_AE.M_y = M_y.permute(0, 2, 3, 1)[0]

test_AE.Mixed = Tester(load_path_Mixed, device)

test_AE.N_f

test_AE.N_loss

test_AE.N_y = N_y.permute(0, 2, 3, 1)[0]

string test_AE.name = 'Billiard_AE'

test_AE.NT = Tester(load_path_NT, device)

test_AE.NT_f

test_AE.NT_loss

test_AE.NT_y = NT_y.permute(0, 2, 3, 1)[0]

int test_AE.number_of_samples = 10

test_AE.RANDOM_ARM_INIT_POSE

test_AE.RANDOM_BALL_INIT_POSE

test_AE.S_f

test_AE.S_loss

test_AE.S_y = S_y.permute(0, 2, 3, 1)[0]

int test_AE.seed = 3

test_AE.subs = AEN.selector.subsample(images_test[k:k+1])

bool test_AE.TEST_TRAINED = True

test_AE.tmp = env.render(mode='rgb_array')

test_AE.x_test = []