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