trtd56 commited on
Commit
793c2d9
1 Parent(s): 6ed9089

Update script.py

Browse files
Files changed (1) hide show
  1. script.py +9 -3
script.py CHANGED
@@ -1,9 +1,15 @@
1
- import os
2
  import pandas as pd
3
 
4
- list1 = os.listdir("../data")
5
  print("################################")
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("trtd56/compe_test", data_files={"test": "test.csv"})
6
+ print(test_dset)
7
+ #DatasetDict({
8
+ # test: Dataset({
9
+ # features: ['id', 'Query', 'FAQ1', 'FAQ2', 'FAQ3'],
10
+ # num_rows: 837
11
+ # })
12
+ #})
13
  print("################################")
14
 
15
  df = pd.DataFrame([(f"testid{i:04}", 0) for i in range(837)], columns=["id", "pred"])