This repository contains Python bindings for working with Nomic Atlas, the world’s most powerful unstructured data interaction platform. Atlas supports datasets from hundreds to tens of millions of points, and supports data modalities ranging from text to image to audio to video.
With Nomic Atlas, you can:
- Generate, store and retrieve embeddings for your unstructured data.
- Find insights in your unstructured data and embeddings all from your web browser.
- Share and present your datasets and data findings to anyone.
Articles Submitted to Arxiv (10/12/2023 - 10/19/2023) |
Historical TikTok Dataset (Indexed on Metadata Descriptions) |
Try the 📓 Colab Demo to get started in Python
Read the 📕 Atlas Docs
Join our 🛖 Discord to start chatting and get help
🗺️ Map of Twitter (5.4 million tweets)
🗺️ Map of StableDiffusion Generations (6.4 million images)
🗺️ Map of NeurIPS Proceedings (16,623 abstracts)
Here are just a few of the features which Atlas offers:
- Organize your text, image, and embedding data
- Create beautiful and shareable maps with or without coding knowledge
- Have easy access to both high-level data structures and individual datapoints
- Search millions of datapoints instantly
- Cluster data into semantic topics
- Tag and clean your dataset
- Deduplicate text, images, video, audio
- Install the Nomic library
pip install nomic- Login or create your Nomic account:
nomic login- Follow the instructions to obtain your access token.
nomic login [token]from nomic import atlas
import numpy as np
# Randomly generate a set of 10,000 high-dimensional embeddings
num_embeddings = 10000
embeddings = np.random.rand(num_embeddings, 256)
# Create Atlas project
project = atlas.map_embeddings(embeddings=embeddings)
print(project)Atlas stores, manages and generates embeddings for your unstructured data.
You can access Atlas latent embeddings (e.g. high dimensional) or the two-dimensional embeddings generated for web display.
# Access your Atlas map and download your embeddings
map = project.maps[0]
projected_embeddings = map.embeddings.projected
latent_embeddings = map.embeddings.latentprint(projected_embeddings)# Response:
id x y
0 9.815330 -8.105308
1 -8.725819 5.980628
2 13.199472 -1.103389
... ... ... ...
print(latent_embeddings)# Response:
n x d numpy.ndarray where n = number of datapoints and d = number of latent dimensions
Atlas automatically organizes your data into topics informed by the latent contents of your embeddings. Visually, these are represented by regions of homogenous color on an Atlas map.
You can access and operate on topics programmatically by using the topics attribute
of an AtlasMap.
# Access your Atlas map
map = project.maps[0]
# Access a pandas dataframe associating each datum on your map to their topics at each topic depth.
topic_df = map.topics.df
print(map.topics.df)Response:
id topic_depth_1 topic_depth_2
0 Oil Prices mergers and acquisitions
1 Iraq War Trial of Thatcher
2 Oil Prices Economic Growth
... ... ... ...
9997 Oil Prices Economic Growth
9998 Baseball Giambi's contract
9999 Olympic Gold Medal European Football
Use Atlas to automatically find nearest neighbors in your vector database.
# Load map and perform vector search for the five nearest neighbors of datum with id "my_query_point"
map = project.maps[0]
with project.wait_for_project_lock():
neighbors, _ = map.embeddings.vector_search(ids=['my_query_point'], k=5)
# Return similar data points
similar_datapoints = project.get_data(ids=neighbors[0])
print(similar_datapoints)Response:
Original query point:
"Intel abandons digital TV chip project NEW YORK, October 22 (newratings.com) - Global semiconductor giant Intel Corporation (INTC.NAS) has called off its plan to develop a new chip for the digital projection televisions."
Nearest neighbors:
"Intel awaits government move on expensing options Figuring it's had enough of fighting over options, the chip giant is waiting to see what Congress comes up with."
"Citigroup Takes On Intel The financial services giant takes over non-memory semiconductor chip production."
"Intel Seen Readying New Wi-Fi Chips SAN FRANCISCO (Reuters) - Intel Corp. this week is expected to introduce a chip that adds support for a relatively obscure version of Wi-Fi, analysts said on Monday, in a move that could help ease congestion on wireless networks."
"Intel pledges to bring Itanic down to Xeon price-point EM64T a stand-in until the real anti-AMD64 kit arrives"
Detailed documentation can be found at https://docs.nomic.ai.
Atlas is developed by the Nomic AI team, which is based in NYC. Nomic also developed and maintains GPT4All, an open-source LLM chatbot ecosystem.
Join the discussion on our 🛖 Discord to ask questions, get help, and chat with others about Atlas, Nomic, GPT4All, and related topics. Our doors are open to enthusiasts of all skill levels.
- Blog: https://blog.nomic.ai/
- Twitter: https://twitter.com/nomic_ai
- Nomic Website: https://home.nomic.ai/
- Atlas Website: https://atlas.nomic.ai/
- GPT4All Website: https://gpt4all.io/index.html
- LinkedIn: https://www.linkedin.com/company/nomic-ai