namngo commited on
Commit
df2f774
·
verified ·
1 Parent(s): 09410d3

Update dataset.py

Browse files
Files changed (1) hide show
  1. dataset.py +12 -12
dataset.py CHANGED
@@ -1,12 +1,12 @@
1
- import pandas as pd
2
- from sklearn.model_selection import train_test_split
3
- from datasets import DatasetDict, Dataset
4
-
5
- # Load dữ liệu từ CSV (hoặc thay bằng đường dẫn file của bạn)
6
- a = pd.read_csv("D:/Pycharm/Project/Project2/Model/datav6.csv")
7
- a=a[['context','question','answers']]
8
- dataset_a = Dataset.from_pandas(a)
9
- # print(dataset_a)
10
- # print(dataset_a["context"]) # Giả sử tập train có cột "pos"
11
-
12
-
 
1
+ import pandas as pd
2
+ from sklearn.model_selection import train_test_split
3
+ from datasets import DatasetDict, Dataset
4
+
5
+ # Load dữ liệu từ CSV (hoặc thay bằng đường dẫn file của bạn)
6
+ a = pd.read_csv("datav6.csv")
7
+ a=a[['context','question','answers']]
8
+ dataset_a = Dataset.from_pandas(a)
9
+ # print(dataset_a)
10
+ # print(dataset_a["context"]) # Giả sử tập train có cột "pos"
11
+
12
+