- Descrição :
Embeddings pré-treinados para busca aproximada do vizinho mais próximo usando a distância do cosseno. Este conjunto de dados consiste em duas divisões:
- 'banco de dados': consiste em 9.990.000 pontos de dados, cada um com recursos: 'embedding' (96 floats), 'index' (int64), 'neighbors' (lista vazia).
- 'test': consiste em 10.000 pontos de dados, cada um possui recursos: 'embedding' (96 floats), 'index' (int64), 'neighbors' (lista de 'index' e 'distance' dos vizinhos mais próximos no banco de dados. )
Página inicial : http://sites.skoltech.ru/compvision/noimi/
Código -fonte:
tfds.nearest_neighbors.deep1b.Deep1b
Versões :
-
1.0.0
(padrão): versão inicial.
-
Tamanho do download :
3.58 GiB
Tamanho do conjunto de dados :
4.46 GiB
Armazenado em cache automaticamente ( documentação ): Não
Divisões :
Dividir | Exemplos |
---|---|
'database' | 9.990.000 |
'test' | 10.000 |
- Estrutura de recursos :
FeaturesDict({
'embedding': Tensor(shape=(96,), dtype=float32),
'index': Scalar(shape=(), dtype=int64),
'neighbors': Sequence({
'distance': Scalar(shape=(), dtype=float32),
'index': Scalar(shape=(), dtype=int64),
}),
})
- Documentação do recurso:
Característica | Classe | Forma | Tipo D | Descrição |
---|---|---|---|---|
RecursosDict | ||||
incorporação | tensor | (96,) | float32 | |
índice | Escalar | int64 | Índice dentro da divisão. | |
vizinhos | Seqüência | Os vizinhos calculados, que estão disponíveis apenas para a divisão de teste. | ||
vizinhos/distância | Escalar | float32 | Distância do vizinho. | |
vizinhos/índice | Escalar | int64 | Índice vizinho. |
Chaves supervisionadas (Consulte
as_supervised
doc ):None
Figura ( tfds.show_examples ): Não compatível.
Exemplos ( tfds.as_dataframe ):
- Citação :
@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}
}