def generate_centroids(n_centroids, n_dims, is_sobol=True):
if is_sobol:
# centroids as sobol sequence
c = sobol.sobol_generate(n_dims, n_centroids)
c = np.array(c)
else:
# centroids as random vectors
c = np.random.rand(n_centroids, n_dims)
c = c.astype(np.float32)
return c
def _501_generate_centroids(n_centroids, n_dims):
c1_path = Pth(
'Breakfast/features_centroids/features_random_%d_centroids.pkl',
(n_centroids, ))
c2_path = Pth(
'Breakfast/features_centroids/features_sobol_%d_centroids.pkl',
(n_centroids, ))
# centroids as random vectors
c1 = np.random.rand(n_centroids, n_dims)
# centroids as sobol sequence
c2 = sobol.sobol_generate(n_dims, n_centroids)
c2 = np.array(c2)
# save centroids
utils.pkl_dump(c1, c1_path)
utils.pkl_dump(c2, c2_path)
def _602_generate_nodes(n_nodes, n_dims):
pass
n1_path = Pth('EPIC-Kitchens/features_centroid/features_random_%d.pkl', (n_nodes,))
n2_path = Pth('EPIC-Kitchens/features_centroid/features_sobol_%d.pkl', (n_nodes,))
# nodes as random vectors
n1 = np.random.rand(n_nodes, n_dims)
# nodes as sobol sequence
n2 = sobol.sobol_generate(n_dims, n_nodes)
n2 = np.array(n2)
print n1.shape
print n2.shape
# save nodes
utils.pkl_dump(n1, n1_path)
utils.pkl_dump(n2, n2_path)
def generate_centroids(n, dim, is_sobol=False):
# nodes as sobol sequence or random vectors
centroids = np.array(sobol.sobol_generate(
dim, n)) if is_sobol else np.random.rand(n, dim)
return centroids