MarkusStoll's picture
layout, spotlight fix, cache fix
bb4dab5
raw
history blame
1.03 kB
import pickle
import datasets
from renumics import spotlight
import os
if __name__ == "__main__":
cache_file = "dataset_cache.pkl"
if os.path.exists(cache_file):
# Load dataset from cache
with open(cache_file, "rb") as file:
dataset = pickle.load(file)
print("Dataset loaded from cache.")
else:
# Load dataset using datasets.load_dataset()
dataset = datasets.load_dataset("renumics/cifar100-enriched", split="train")
print("Dataset loaded using datasets.load_dataset().")
# Save dataset to cache
with open(cache_file, "wb") as file:
pickle.dump(dataset, file)
print("Dataset saved to cache.")
df = dataset.to_pandas()
df_show = df.drop(columns=['embedding', 'probabilities'])
spotlight.show(df_show.sample(5000, random_state=1), port=7860, host="0.0.0.0",
dtype={"image": spotlight.Image, "embedding_reduced": spotlight.Embedding}, allow_filebrowsing=False, layout="spotlight-layout.json")