research14 commited on
Commit
691be8c
·
1 Parent(s): ac18062

edited corpus

Browse files
Files changed (1) hide show
  1. run_llm.py +2 -2
run_llm.py CHANGED
@@ -74,13 +74,13 @@ for m in model_mapping.keys():
74
 
75
  s = 0
76
  e = 1000
77
- with open('ptb_corpus/sample_uniform_1k_2.txt', 'r') as f:
78
  selected_idx = f.readlines()
79
  selected_idx = [int(i.strip()) for i in selected_idx][s:e]
80
 
81
 
82
  ptb = []
83
- with open('./ptb_corpus/ptb.jsonl', 'r') as f:
84
  for l in f:
85
  ptb.append(json.loads(l))
86
 
 
74
 
75
  s = 0
76
  e = 1000
77
+ with open('test/sample_uniform_1k_2.txt', 'r') as f:
78
  selected_idx = f.readlines()
79
  selected_idx = [int(i.strip()) for i in selected_idx][s:e]
80
 
81
 
82
  ptb = []
83
+ with open('test/ptb.jsonl', 'r') as f:
84
  for l in f:
85
  ptb.append(json.loads(l))
86