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)))
