sps44's picture
new spotlight version
4b062f5
raw
history blame
998 Bytes
import pickle
import datasets
from renumics import spotlight
import os
import pandas as pd
from renumics.spotlight.layouts import compare_classification
if __name__ == "__main__":
cache_file_issues="embedding_issues.pkl"
# Load dataset using datasets.load_dataset()
ds = datasets.load_dataset("renumics/emodb-enriched", split="all")
print("Dataset loaded using datasets.load_dataset().")
with open(cache_file_issues, "rb") as issue_file:
issues = pickle.load(issue_file)
while True:
layout = compare_classification(label='gender', model1_prediction='m1_gender_prediction', model1_embedding='m1_embedding', model1_correct='m1_correct', model2_prediction='m2_gender_prediction', model2_embedding='m2_embedding', model2_correct='m2_correct', inspect={'audio': spotlight.Audio})
view = spotlight.show(ds, issues=issues, port=7860, host="0.0.0.0", layout=layout,
allow_filebrowsing=False)
view.close()