modules/text2vec-huggingface/clients/bert_embeddings.go

//                           _       _
// __      _____  __ ___   ___  __ _| |_ ___
// \ \ /\ / / _ \/ _` \ \ / / |/ _` | __/ _ \
//  \ V  V /  __/ (_| |\ V /| | (_| | ||  __/
//   \_/\_/ \___|\__,_| \_/ |_|\__,_|\__\___|
//
//  Copyright © 2016 - 2024 Weaviate B.V. All rights reserved.
//
//  CONTACT: [email protected]
//

package clients

import "errors"

type bertEmbeddingsDecoder struct{}

func newBertEmbeddingsDecoder() *bertEmbeddingsDecoder {
	return &bertEmbeddingsDecoder{}
}

func (d bertEmbeddingsDecoder) calculateVector(embeddings [][]float32) ([]float32, error) {
	if len(embeddings) > 0 {
		vectorLen := len(embeddings[0])
		sumEmbeddings := make([]float32, vectorLen)
		embeddingsLen := len(embeddings)
		var sum float32
		for i := 0; i < vectorLen; i++ {
			sum = 0
			for j := 0; j < embeddingsLen; j++ {
				sum += embeddings[j][i]
			}
			sumEmbeddings[i] = sum
		}
		for i := range sumEmbeddings {
			sumEmbeddings[i] = sumEmbeddings[i] / float32(embeddingsLen)
		}
		return sumEmbeddings, nil
	}
	return nil, errors.New("missing embeddings")
}