summaryrefslogtreecommitdiff
path: root/recital_dataset.py
diff options
context:
space:
mode:
Diffstat (limited to 'recital_dataset.py')
-rw-r--r--recital_dataset.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/recital_dataset.py b/recital_dataset.py
index 73a9aca..a142c92 100644
--- a/recital_dataset.py
+++ b/recital_dataset.py
@@ -3,7 +3,7 @@ import random
import json
# 读取CSV文件
-df = pd.read_csv('recitalMachine_dataset.csv') # 使用制表符分隔
+df = pd.read_csv('./recitalMachine/recitalMachine_dataset.csv') # 使用制表符分隔
df['length_of_answer'] = df['answer'].apply(lambda x: len(str(x).split(',')))
# 按difficulty分组并创建子DataFrame列表