switch to new spotlight version

Dockerfile

@@ -2,13 +2,13 @@ FROM python:3.9
 
 WORKDIR /code
 ENV HOME=/code
-COPY ./renumics_spotlight-1.3.0.post109+71fb2cb-py3-none-any.whl /code/renumics_spotlight-1.3.0.post109+71fb2cb-py3-none-any.whl
+COPY ./renumics_spotlight-1.4.0.post18+cd16f7b-py3-none-any.whl /code/renumics_spotlight-1.4.0.post18+cd16f7b-py3-none-any.whl
 # RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
 RUN apt install curl
 RUN pip install pip -U
 
-RUN pip install renumics_spotlight-1.3.0.post109+71fb2cb-py3-none-any.whl
+RUN pip install renumics_spotlight-1.4.0.post18+cd16f7b-py3-none-any.whl
 
 
 RUN pip install pyarrow datasets 

emodb_embedding_issues.pkl → embedding_issues.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5f4a5b1d01f9e233e28b90b00cf5c797cc28b469ea8ee266c76db02ed436c44f
-size 1607
+oid sha256:f14d05f9c09ce14e208c72c463ee745de34be9ba6c59f0af9e7ccf0295baf7f0
+size 1647

emodb_full_results.parquet

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:0a0abe6f42c0bc929838e1f56ac956b34b7d75ff4c45e01d3a53a5459d8fa7fa
-size 11141947

layout-model-comparison.json

@@ -1 +0,0 @@
-{"orientation":"horizontal","children":[{"kind":"split","weight":22.443871668302805,"orientation":"vertical","children":[{"kind":"tab","weight":50,"children":[{"kind":"widget","name":"Table","type":"table","config":{"tableView":"selected","visibleColumns":null,"sorting":null,"orderByRelevance":false}}]},{"kind":"tab","weight":50,"children":[{"kind":"widget","name":"Similarity Map","type":"similaritymap","config":{"placeBy":null,"reductionMethod":null,"colorBy":"m1_gender_prediction_error","sizeBy":null,"filter":false,"umapNNeighbors":20,"umapMetric":null,"umapMinDist":0.15,"pcaNormalization":null,"umapMenuLocalGlobalBalance":null,"umapMenuIsAdvanced":false}}]}]},{"kind":"split","weight":19.08256196026494,"orientation":"vertical","children":[{"kind":"split","weight":4.406531194985268,"orientation":"horizontal","children":[{"kind":"tab","weight":50,"children":[{"kind":"widget","name":"Accuracy Filtered vs. Selected","type":"Metric","config":{"config":{"metric":"accuracy","columns":{"X":"m1_gender_prediction_error"}}}}]},{"kind":"tab","weight":50,"children":[{"kind":"widget","name":"F1-Score Filtered vs. Selected","type":"Metric","config":{"config":{"metric":"F1","columns":{"X":"gender_bool","Y":"m1_gender_pred_bool"}}}}]}]},{"kind":"tab","weight":12.861339016257817,"children":[{"kind":"widget","name":"Confusion Matrix","type":"ConfusionMatrix","config":{"xColumn":"gender","yColumn":"m1_gender_prediction"}}]},{"kind":"tab","weight":12.861339016257817,"children":[{"kind":"widget","name":"Issues (from Sliceguard)","type":"IssuesWidget"}]}]},{"kind":"split","weight":17.45935648231709,"orientation":"vertical","children":[{"kind":"tab","weight":50,"children":[{"kind":"widget","name":"Gender vs. Emotion","type":"ConfusionMatrix","config":{"xColumn":"gender","yColumn":"emotion"}}]},{"kind":"tab","weight":50,"children":[{"kind":"widget","name":"Gender vs. Age","type":"ConfusionMatrix","config":{"xColumn":"gender","yColumn":"age"}}]}]},{"kind":"tab","weight":16.908583738243667,"children":[{"kind":"widget","name":"Inspector","type":"inspector","config":{"views":[{"view":"SpectrogramView","columns":["audio"],"name":"view","key":"07aaaf22-1586-453a-b9cb-674ea68f1079"},{"view":"AudioView","key":"5dKVmNM6c2fijG2GAPAKhD","name":"audio","columns":["audio"]},{"view":"ScalarView","columns":["gender"],"name":"view","key":"ac749b5f-98c5-47da-9026-df3134fd9ffb"},{"view":"TextLens","columns":["m1_gender_prediction_string"],"name":"view","key":"2c85598a-750e-4186-94aa-d311dd7aec37"}],"visibleColumns":4}}]}]}

layout-model-debug.json

@@ -1 +0,0 @@
-{"orientation":"horizontal","children":[{"kind":"split","weight":22.1951801110117,"orientation":"vertical","children":[{"kind":"tab","weight":50,"children":[{"kind":"widget","name":"Table","type":"table","config":{"tableView":"full","visibleColumns":null,"sorting":null,"orderByRelevance":false}}]},{"kind":"tab","weight":50,"children":[{"kind":"widget","name":"Embedding on emotion model","type":"similaritymap","config":{"placeBy":null,"reductionMethod":null,"colorBy":null,"sizeBy":null,"filter":false,"umapNNeighbors":20,"umapMetric":null,"umapMinDist":0.15,"pcaNormalization":null,"umapMenuLocalGlobalBalance":null,"umapMenuIsAdvanced":false}}]}]},{"kind":"split","weight":15.587276091496047,"orientation":"vertical","children":[{"kind":"split","weight":11.751097808946518,"orientation":"horizontal","children":[{"kind":"tab","weight":50,"children":[{"kind":"widget","name":"Accuracy","type":"Metric","config":{"config":{"metric":"accuracy","columns":{"X":"gender_bool","y":"m1_gender_pred_bool"}}}}]},{"kind":"tab","weight":50,"children":[{"kind":"widget","name":"F1","type":"Metric","config":{"config":{"metric":"F1","columns":{"X":"gender_bool","Y":"m1_gender_pred_bool"}}}}]}]},{"kind":"tab","weight":45.964471075702306,"children":[{"kind":"widget","name":"Confusion Matrix","type":"ConfusionMatrix","config":{"xColumn":"gender","yColumn":"m1_gender_prediction_string"}}]},{"kind":"tab","weight":42.28443111535119,"children":[{"kind":"widget","name":"Issues from Sliceguard","type":"IssuesWidget"}]}]},{"kind":"split","weight":11.561423326142764,"orientation":"vertical","children":[{"kind":"tab","weight":50,"children":[{"kind":"widget","name":"Gender vs. Emotion","type":"ConfusionMatrix","config":{"xColumn":"gender","yColumn":"emotion"}}]},{"kind":"tab","weight":50,"children":[{"kind":"widget","name":"Confusion Matrix (2)","type":"ConfusionMatrix","config":{"xColumn":"gender","yColumn":"age"}}]}]},{"kind":"tab","weight":16.634733093151677,"children":[{"kind":"widget","name":"Inspector","type":"inspector","config":{"views":[{"view":"AudioView","key":"tY8GHhvKXUvWMumcC6aEK4","name":"audio","columns":["audio"]},{"view":"ScalarView","columns":["gender"],"name":"view","key":"464686a8-db0d-40a5-a31d-38323f66e99a"},{"view":"TextLens","columns":["m1_gender_prediction_string"],"name":"view","key":"9ced1617-7d59-4abf-acb0-c9a8ed393ebf"},{"view":"SpectrogramView","columns":["audio"],"name":"view","key":"95aead50-93b4-4a80-a9b2-38a5ba1521f6"}],"visibleColumns":4}}]}]}

emodb_metadata_issues.pkl → metadata_issues.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6ae74f1f53f539a5ea1418210e8126bf6849b3b2a5366b170f9558893fca9953
-size 3695
+oid sha256:a984f1de18e34614d5e5383a5c0476723588171d8f01fcfcdd2cdfbe62b7a3d1
+size 3869

prepare.py

@@ -3,21 +3,21 @@ import datasets
 import os
 import pandas as pd
 
+from datasets import load_dataset_builder
+
 if __name__ == "__main__":
-    cache_file = "dataset_cache.parquet"
-    if os.path.exists(cache_file):
+    
+    builder = load_dataset_builder('renumics/emodb-enriched')
+
+    
+    if os.path.exists(builder._cache_dir):
         # Load dataset from cache
-        df = pd.read_parquet(cache_file)
+        ds = datasets.load_dataset('renumics/emodb-enriched', split='all')
         print("Dataset loaded from cache.")
     else:
         # Load dataset using datasets.load_dataset()
-        dataset = datasets.load_dataset("renumics/emodb", split="all")
+        ds = datasets.load_dataset('renumics/emodb-enriched', split='all')
         print("Dataset loaded using datasets.load_dataset().")
         
-        df = dataset.to_pandas()      
-
-        # Save dataset to cache
-        df.to_parquet(cache_file)
-
-        print("Dataset saved to cache.")
+      
 

renumics_spotlight-1.3.0.post109+71fb2cb-py3-none-any.whl → renumics_spotlight-1.4.0.post18+cd16f7b-py3-none-any.whl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:124c97722387cede2c3fa374e68862bcf65dc0513346e7a09bba73031b8985c1
-size 2525380
+oid sha256:a3ec724a28fdd2e88bcad9de202781629bef33ee9498703583161b5a513ad4eb
+size 2936788

run.py

@@ -3,41 +3,26 @@ import datasets
 from renumics import spotlight
 import os
 import pandas as pd
+from renumics.spotlight.layouts import debug_classification
+
+
 
 if __name__ == "__main__":
-    cache_file = "dataset_cache.parquet"
-    cache_file_enrichment="emodb_full_results.parquet"
-    cache_file_issues="emodb_metadata_issues.pkl"
-    
-    if os.path.exists(cache_file):
-        # Load dataset from cache
-        df_db = pd.read_parquet(cache_file) 
-        print("Dataset loaded from cache.")
-    else:
-        # Load dataset using datasets.load_dataset()
-        dataset = datasets.load_dataset("renumics/emodb", split="all")
-        print("Dataset loaded using datasets.load_dataset().")
-        
-        df_db = dataset.to_pandas()  
 
-        # Save dataset to cache
-        df_db.to_parquet(cache_file)
+    cache_file_issues="metadata_issues.pkl"
+   
+    # Load dataset using datasets.load_dataset()
+    ds = datasets.load_dataset("renumics/emodb-enriched", split="all")
+    print("Dataset loaded using datasets.load_dataset().")
 
-        print("Dataset saved to cache.")
-    
-    df=pd.read_parquet(cache_file_enrichment) 
-    
-    df['audio'] = df_db['audio']
      
     with open(cache_file_issues, "rb") as issue_file:
         issues = pickle.load(issue_file)
 
 
     while True:
-        dtypes = dtype={"audio": spotlight.Audio, 
-                'm1_embedding':spotlight.Embedding, 
-                'm2_embedding':spotlight.Embedding, 
-                'embedding_emotion':spotlight.Embedding}
-        view = spotlight.show(df, issues=issues, port=7860, host="0.0.0.0", layout="layout-model-debug.json",
-                    dtype=dtypes, allow_filebrowsing=False)
+        layout = debug_classification(label='gender', prediction='m1_gender_prediction', embedding='m1_embedding', features=['age', 'emotion'], inspect={'audio': spotlight.Audio})
+        
+        view = spotlight.show(ds, issues=issues, port=7860, host="0.0.0.0", layout=layout,
+                    allow_filebrowsing=False)
         view.close()