Finance BGE Embeddings

Description

The BGE embedding model was trained on a mix of different datasets. We used public data and in-house annotated documents.

Predicted Entities

Copy S3 URI

How to use

embeddings  =  nlp.BGEEmbeddings.pretrained("finance_bge_base_embeddings","en","finance/models")\
    .setInputCols("document")\ 
    .setOutputCol("embeddings")

Model Information

Model Name: finance_bge_base_embeddings
Compatibility: Finance NLP 1.0.0+
License: Licensed
Edition: Official
Input Labels: [document]
Output Labels: [sentence_embeddings]
Language: en
Size: 400.6 MB

References

Public data and in-house annotated documents