hf challam robertabase without date (checkpoint 395000) epoch1

2021-12-13 13:22:51 +01:00 · 2021-12-13 13:22:51 +01:00 · 0334428ec5
commit 0334428ec5
parent 7bcf8513d1
2 changed files with 8495 additions and 10516 deletions
--- a/hf_challam_roberta_base/04_predict_from_file.py
+++ b/hf_challam_roberta_base/04_predict_from_file.py
@ -34,7 +34,7 @@ with open('scalers.pickle', 'rb') as f_scaler:
 def predict(dataset, out_f):
    outputs = []

-    for sample in tqdm(eval_dataset_full[1:]):
+    for sample in tqdm(dataset[1:]):
        y, t = sample.split('\t')
        t = t.rstrip()

--- a/test-A/out.tsv
+++ b/test-A/out.tsv