przesiać 1 m

  • Opis :

Wstępnie przeszkolone osadzania do przybliżonego wyszukiwania najbliższego sąsiada przy użyciu odległości euklidesowej. Ten zbiór danych składa się z dwóch podziałów:

  1. „baza danych”: składa się z 1 000 000 punktów danych, każdy ma funkcje: „osadzanie” (128 zmiennych), „indeks” (int64), „sąsiedzi” (pusta lista).
  2. „test”: składa się z 10 000 punktów danych, każdy ma funkcje: „osadzanie” (128 float), „indeks” (int64), „sąsiedzi” (lista „indeksu” i „odległości” najbliższych sąsiadów w bazie danych. )
Podział Przykłady
'database' 1 000 000
'test' 10 000
  • Struktura funkcji :
FeaturesDict({
    'embedding': Tensor(shape=(128,), dtype=float32),
    'index': Scalar(shape=(), dtype=int64, description=Index within the split.),
    'neighbors': Sequence({
        'distance': Scalar(shape=(), dtype=float32, description=Neighbor distance.),
        'index': Scalar(shape=(), dtype=int64, description=Neighbor index.),
    }),
})
  • Dokumentacja funkcji :
Funkcja Klasa Kształt Typ D Opis
FunkcjeDykt
osadzanie Napinacz (128,) pływak32
indeks Skalarny int64 Indeks w ramach podziału.
sąsiedzi Sekwencja Obliczeni sąsiedzi, którzy są dostępni tylko w przypadku podziału testowego.
sąsiedzi/odległość Skalarny pływak32 Odległość sąsiada.
sąsiedzi/indeks Skalarny int64 Indeks sąsiada.
  • Cytat :
@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}
}