Update script.py
Browse files
script.py
CHANGED
@@ -1,9 +1,15 @@
|
|
1 |
-
import os
|
2 |
import pandas as pd
|
3 |
|
4 |
-
list1 = os.listdir("../data")
|
5 |
print("################################")
|
6 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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"])
|