def load_ground_truth(filename, file_format, dimension = 1000, nuse = 10000, offset = 0): (feature_vecs, actual_nuse) = yutils.load_vectors_fmt(filename, file_format, dimension, nuse, offset, verbose = True) feature_vecs = yael.ivec_to_numpy(feature_vecs, int(actual_nuse) * dimension) feature_vecs = feature_vecs.reshape((int(actual_nuse), dimension)) return (feature_vecs, actual_nuse)
def ivecs_read(filename): (fvecs, n, d) = yael.ivecs_new_read(filename) if n == -1: raise IOError("could not read " + filename) elif n == 0: d = 0 ivecs = yael.ivec.acquirepointer(fvecs) # TODO find a way to avoid copy a = yael.ivec_to_numpy(fvecs, n * d) return a.reshape((n, d))
def load_ground_truth(filename, file_format, dimension=1000, nuse=10000, offset=0): (feature_vecs, actual_nuse) = yutils.load_vectors_fmt(filename, file_format, dimension, nuse, offset, verbose=True) feature_vecs = yael.ivec_to_numpy(feature_vecs, int(actual_nuse) * dimension) feature_vecs = feature_vecs.reshape((int(actual_nuse), dimension)) return (feature_vecs, actual_nuse)