Ollama嵌入¶

Ollama是一个用于学习嵌入表示的Python库。嵌入表示是将对象映射到连续向量空间的技术，通常用于自然语言处理和推荐系统中。Ollama库提供了一种简单而灵活的方法来训练和使用嵌入表示。

如果您在colab上打开这个笔记本，您可能需要安装LlamaIndex 🦙。

In [ ]:

%pip install llama-index-embeddings-ollama

In [ ]:

!pip install llama-index

In [ ]:

from llama_index.embeddings.ollama import OllamaEmbedding

ollama_embedding = OllamaEmbedding(
    model_name="llama2",
    base_url="http://localhost:11434",
    ollama_additional_kwargs={"mirostat": 0},
)

pass_embedding = ollama_embedding.get_text_embedding_batch(
    ["This is a passage!", "This is another passage"], show_progress=True
)
print(pass_embedding)

query_embedding = ollama_embedding.get_query_embedding("Where is blue?")
print(query_embedding)