def __init__(self): self.images, features = _load_small_norb_chunks( SMALLNORB_TEMPLATE, SMALLNORB_CHUNKS) self.factor_sizes = [5, 10, 9, 18, 6] # Instances are not part of the latent space. self.latent_factor_indices = [0, 2, 3, 4] self.num_total_factors = features.shape[1] self.index = util.StateSpaceAtomIndex(self.factor_sizes, features) self.state_space = util.SplitDiscreteStateSpace( self.factor_sizes, self.latent_factor_indices)
def __init__(self): self.factor_sizes = [4, 24, 183] features = cartesian([np.array(list(range(i))) for i in self.factor_sizes]) self.latent_factor_indices = [0, 1, 2] self.num_total_factors = features.shape[1] self.index = util.StateSpaceAtomIndex(self.factor_sizes, features) self.state_space = util.SplitDiscreteStateSpace(self.factor_sizes, self.latent_factor_indices) self.data_shape = [64, 64, 3] self.images = self._load_data()
def test(self): features = np.array([[0, 1], [1, 0], [1, 1], [0, 0]], dtype=np.int64) state_space_atom_index = util.StateSpaceAtomIndex([2, 2], features) self.assertAllEqual(state_space_atom_index.features_to_index(features), list(range(4)))