What is Embeddinghub?
Embeddinghub is a database built for machine learning embeddings. It is built with four goals in mind.
- Store embeddings durably and with high availability
- Allow for approximate nearest neighbor operations
- Enable other operations like partitioning, sub-indices, and averaging
- Manage versioning, access control, and rollbacks painlessly
Features
- Supported Operations: Run approximate nearest neighbor lookups, average multiple embeddings, partition tables (spaces), cache locally while training, and more.
- Storage: Store and index billions vectors embeddings from our storage layer.
- Versioning: Create, manage, and rollback different versions of your embeddings.
- Access Control: Encode different business logic and user management directly into Embeddinghub.
- Monitoring: Keep track of how embeddings are being used, latency, throughput, and feature drift over time.
What is an Embedding?
Embeddings are dense numerical representations of real-world objects and relationships, expressed as a vector. The vector space quantifies the semantic similarity between categories. Embedding vectors that are close to each other are considered similar. Sometimes, they are used directly for “Similar items to this” section in an e-commerce store. Other times, embeddings are passed to other models. In those cases, the model can share learnings across similar items rather than treating them as two completely unique categories, as is the case with one-hot encodings. For this reason, embeddings can be used to accurately represent sparse data like clickstreams, text, and e-commerce purchases as features to downstream models.
Further Reading
- Read up on common embeddings use cases, like recommender systems, nearest neighbor, and natural language processing in our docs.
- The Definitive Guide to Embeddings
Getting Started
Step 1: Install Embeddinghub client
Install the Python SDK via pip
pip install embeddinghub
Step 2: Deploy Docker container ( optional )
The Embeddinghub client can be used without a server. This is useful when using embeddings in a research environment where a database server is not necessary. If that’s the case for you, skip ahead to the next step.
Otherwise, we can use this docker command to run Embeddinghub locally and to map the container's main port to our host's port.
docker run featureformcom/embeddinghub -p 7462:7462
Step 3: Initialize Python Client
If you deployed a docker container, you can initialize the python client.
import embeddinghub as eh
hub = eh.connect(eh.Config())
Otherwise, you can use a LocalConfig to store and index embeddings locally.
hub = eh.connect(eh.LocalConfig("data/"))
Step 4: Create a Space
Embeddings are written and retrieved from Spaces. When creating a Space we must also specify a version, otherwise a default version is used.
space = hub.create_space("quickstart", dims=3)
Step 5: Upload Embeddings
We will create a dictionary of three embeddings and upload them to our new quickstart space.
embeddings = {
"apple": [1, 0, 0],
"orange": [1, 1, 0],
"potato": [0, 1, 0],
"chicken": [-1, -1, 0],
}
space.multiset(embeddings)
Step 6: Get nearest neighbors
Now we can compare apples to oranges and get the nearest neighbors.
neighbors = space.nearest_neighbors(key="apple", num=2)
print(neighbors)
Contributing
- To contribute to Embeddinghub, please check out Contribution docs.
- Welcome to our awesome community, please join our Slack community.
Report Issues
Please help us by reporting any issues you may have while using Embeddinghub.
![[Bug]: Inability to do time-based join, and potentially to name time column different than 'ts'](https://avatars.githubusercontent.com/u/125003?v=4)
![[Feature] Docker Container Quickstart](https://avatars.githubusercontent.com/u/15040917?v=4)
186 Dec 27, 2022
815 Dec 29, 2022
4 Sep 21, 2021
99 Dec 27, 2022
602 Jan 03, 2023
59 Nov 25, 2022
7 Dec 27, 2021
39 Dec 25, 2022
1 Aug 18, 2022
24 Dec 17, 2022
4 May 10, 2022
20k Jan 05, 2023
8 Oct 03, 2022
1 Jan 05, 2022
93 Nov 28, 2022
51 Dec 29, 2022
55 Dec 09, 2022
83 Dec 24, 2022
27 Dec 21, 2022
6.9k Dec 29, 2022