sps44 commited on
Commit
30747fb
1 Parent(s): 8157d79

test split with layout

Browse files
Files changed (2) hide show
  1. prepare.py +1 -1
  2. run.py +2 -2
prepare.py CHANGED
@@ -11,7 +11,7 @@ if __name__ == "__main__":
11
  print("Dataset loaded from cache.")
12
  else:
13
  # Load dataset using datasets.load_dataset()
14
- dataset = datasets.load_dataset("renumics/cifar100-enriched", split="train")
15
  print("Dataset loaded using datasets.load_dataset().")
16
 
17
  df = dataset.to_pandas()
 
11
  print("Dataset loaded from cache.")
12
  else:
13
  # Load dataset using datasets.load_dataset()
14
+ dataset = datasets.load_dataset("renumics/cifar100-enriched", split="test")
15
  print("Dataset loaded using datasets.load_dataset().")
16
 
17
  df = dataset.to_pandas()
run.py CHANGED
@@ -15,7 +15,7 @@ if __name__ == "__main__":
15
  print("Dataset loaded from cache.")
16
  else:
17
  # Load dataset using datasets.load_dataset()
18
- dataset = datasets.load_dataset("renumics/cifar100-enriched", split="train")
19
  print("Dataset loaded using datasets.load_dataset().")
20
 
21
  df = dataset.to_pandas()
@@ -35,6 +35,6 @@ if __name__ == "__main__":
35
  #df = dataset.to_pandas()
36
  df_show = df.drop(columns=['embedding', 'probabilities'])
37
  while True:
38
- view = spotlight.show(df_show, issues=issues, port=7860, host="0.0.0.0",
39
  dtype={"image": spotlight.Image, "embedding_reduced": spotlight.Embedding}, allow_filebrowsing=False)
40
  view.close()
 
15
  print("Dataset loaded from cache.")
16
  else:
17
  # Load dataset using datasets.load_dataset()
18
+ dataset = datasets.load_dataset("renumics/cifar100-enriched", split="test")
19
  print("Dataset loaded using datasets.load_dataset().")
20
 
21
  df = dataset.to_pandas()
 
35
  #df = dataset.to_pandas()
36
  df_show = df.drop(columns=['embedding', 'probabilities'])
37
  while True:
38
+ view = spotlight.show(df_show, issues=issues, port=7860, host="0.0.0.0", layout="sliceline-layout.json"
39
  dtype={"image": spotlight.Image, "embedding_reduced": spotlight.Embedding}, allow_filebrowsing=False)
40
  view.close()