sps44 commited on
Commit
11197b8
·
1 Parent(s): 3b60ca7
Files changed (2) hide show
  1. prepare.py +1 -1
  2. run.py +1 -1
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/emodb", split="test")
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/emodb", split="all")
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/emodb", split="test")
19
  print("Dataset loaded using datasets.load_dataset().")
20
 
21
  df_db = dataset.to_pandas()
 
15
  print("Dataset loaded from cache.")
16
  else:
17
  # Load dataset using datasets.load_dataset()
18
+ dataset = datasets.load_dataset("renumics/emodb", split="all")
19
  print("Dataset loaded using datasets.load_dataset().")
20
 
21
  df_db = dataset.to_pandas()