StarSpace
Learning embeddings for classification, retrieval and ranking
...Instead of focusing on one task, StarSpace supports multi-task and multi-domain setups—for instance, you can train embeddings so that textual queries match item descriptions, sentences map to labels, or users align with liked items in the same embedding space. The training objective is contrastive: for a given query embedding, positive and negative examples are sampled and the model is optimized to score positive higher than negatives. The library supports a variety of tasks (text classification, nearest-neighbor search, recommendation, entity linking) with simple configuration. ...