saringan1m

  • Deskripsi :

Penyematan terlatih untuk perkiraan pencarian tetangga terdekat menggunakan jarak Euclidean. Dataset ini terdiri dari dua bagian:

  1. 'database': terdiri dari 1.000.000 titik data, masing-masing memiliki fitur: 'embedding' (128 float), 'index' (int64), 'neighbors' (daftar kosong).
  2. 'test': terdiri dari 10.000 titik data, masing-masing memiliki fitur: 'embedding' (128 float), 'index' (int64), 'neighbors' (daftar 'index' dan 'distance' tetangga terdekat di database. )
Membelah Contoh
'database' 1.000.000
'test' 10.000
  • Struktur fitur :
FeaturesDict({
    'embedding': Tensor(shape=(128,), dtype=float32),
    'index': Scalar(shape=(), dtype=int64),
    'neighbors': Sequence({
        'distance': Scalar(shape=(), dtype=float32),
        'index': Scalar(shape=(), dtype=int64),
    }),
})
  • Dokumentasi fitur :
Fitur Kelas Membentuk Dtype Keterangan
fiturDict
penyematan Tensor (128,) float32
indeks Skalar int64 Indeks dalam pemisahan.
tetangga Urutan Tetangga yang dihitung, yang hanya tersedia untuk test split.
tetangga/jarak Skalar float32 Jarak tetangga.
tetangga/indeks Skalar int64 Indeks tetangga.
  • Kutipan :
@article{jegou2010product,
  title={Product quantization for nearest neighbor search},
  author={Jegou, Herve and Douze, Matthijs and Schmid, Cordelia},
  journal={IEEE transactions on pattern analysis and machine intelligence},
  volume={33},
  number={1},
  pages={117--128},
  year={2010},
  publisher={IEEE}
}