tamis1m

  • Description :

Intégrations pré-entraînées pour la recherche approximative du voisin le plus proche en utilisant la distance euclidienne. Cet ensemble de données se compose de deux divisions :

  1. « base de données » : se compose de 1 000 000 de points de données, chacun ayant des fonctionnalités : « intégration » (128 flottants), « index » (int64), « voisins » (liste vide).
  2. 'test' : se compose de 10 000 points de données, chacun ayant des fonctionnalités : 'intégration' (128 flottants), 'index' (int64), 'voisins' (liste des 'index' et 'distance' des voisins les plus proches dans la base de données. )
Diviser Exemples
'database' 1 000 000
'test' 10 000
  • Structure des fonctionnalités :
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.),
    }),
})
  • Documentation des fonctionnalités :
Fonctionnalité Classe Forme Type D Description
FonctionnalitésDict
intégration Tenseur (128,) flotteur32
indice Scalaire int64 Index dans la division.
voisins Séquence Les voisins calculés, disponibles uniquement pour la division de test.
voisins/distance Scalaire flotteur32 Distance du voisin.
voisins/index Scalaire int64 Indice de voisin.
  • Citation :
@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}
}