Spaces:
Runtime error
Runtime error
ronald
commited on
Commit
·
4ebdc9c
1
Parent(s):
17201a4
init
Browse files- local_coh_ppl.py +1 -0
local_coh_ppl.py
CHANGED
|
@@ -211,6 +211,7 @@ class LocalCohPPL(evaluate.Measurement):
|
|
| 211 |
)
|
| 212 |
|
| 213 |
# tokenize by sentence
|
|
|
|
| 214 |
for pred in batch_sents:
|
| 215 |
ss = pred.split("\n")
|
| 216 |
sslens = [len(tokenizer(y,add_special_tokens=False,padding=False).input_ids) for y in ss]
|
|
|
|
| 211 |
)
|
| 212 |
|
| 213 |
# tokenize by sentence
|
| 214 |
+
sent_tok_lens = []
|
| 215 |
for pred in batch_sents:
|
| 216 |
ss = pred.split("\n")
|
| 217 |
sslens = [len(tokenizer(y,add_special_tokens=False,padding=False).input_ids) for y in ss]
|