python 读取 fvecs 数据集

1
2
3
4
5
6
7
8
9
10
11
import numpy as np


def (fname):
a = np.fromfile(fname, dtype='int32')
d = a[0]
return a.reshape(-1, d + 1)[:, 1:].copy()


def fvecs_read(fname):
return ivecs_read(fname).view('float32')