profond1b

  • Description :

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

  1. « base de données » : se compose de 9 990 000 points de données, chacun ayant des fonctionnalités : « intégration » (96 flottants), « index » (int64), « voisins » (liste vide).
  2. 'test' : se compose de 10 000 points de données, chacun ayant des fonctionnalités : 'intégration' (96 flottants), 'index' (int64), 'voisins' (liste des 'index' et 'distance' des voisins les plus proches dans la base de données. )
Diviser Exemples
'database' 9 990 000
'test' 10 000
  • Structure des fonctionnalités :
FeaturesDict({
    'embedding': Tensor(shape=(96,), 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 (96,) 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 :
@inproceedings{babenko2016efficient,
  title={Efficient indexing of billion-scale datasets of deep descriptors},
  author={Babenko, Artem and Lempitsky, Victor},
  booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
  pages={2055--2063},
  year={2016}
}