trtd56 commited on
Commit
6334817
1 Parent(s): b8021e2

Update script.py

Browse files
Files changed (1) hide show
  1. script.py +3 -3
script.py CHANGED
@@ -1,9 +1,9 @@
1
  import pandas as pd
2
 
3
  print("################################")
4
- import os
5
- list1 = os.listdir("/data")
6
- print(list1)
7
  print("################################")
8
 
9
  df = pd.DataFrame([(f"testid{i:04}", 0) for i in range(837)], columns=["id", "pred"])
 
1
  import pandas as pd
2
 
3
  print("################################")
4
+ from datasets import load_dataset
5
+ test_dset = load_dataset("ai-shift/compe_test", data_files={"test": "test.csv"})
6
+ print(test_dset)
7
  print("################################")
8
 
9
  df = pd.DataFrame([(f"testid{i:04}", 0) for i in range(837)], columns=["id", "pred"])