mariabda2 / amazon-sagemaker-nlu-search Goto Github PK

This repository guides users through creating a NLU based product search using Amazon SageMaker and Amazon Elasticsearch service

we have used pre-trained BERT model(distilbert-base-nli-stsb-mean-tokens) from sentence-transformers to generate fixed 768 length sentence embedding on Multi-modal Corpus of Fashion Images from feidegger, a zalandoresearch dataset. Then those feature vectors is imported in Amazon ES KNN Index as a reference.

When we present a new query text/sentence, it's computing the related embedding from Amazon SageMaker hosted BERT model and query Amazon ES KNN index to find similar text/sentence and corresponds to the actual product image which is stored in Amazon S3

This library is licensed under the MIT-0 License. See the LICENSE file.