SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix quality due to ruff release

Browse Source
This commit is contained in:
Sylvain
2023-03-22 20:45:08 -04:00
parent 73fdc8c5b4
commit ef28df0572
28 changed files with 40 additions and 58 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/legacy/pytorch-lightning/run_glue.py
View File

@@ -124,7 +124,7 @@ class GLUETransformer(BaseTransformer):
results = {**{"val_loss": val_loss_mean}, **compute_metrics(self.hparams.task, preds, out_label_ids)}
ret = {k: v for k, v in results.items()}
ret = dict(results.items())
ret["log"] = results
return ret, preds_list, out_label_list

4
examples/legacy/pytorch-lightning/run_ner.py
View File

@@ -122,7 +122,7 @@ class NERTransformer(BaseTransformer):
preds = np.argmax(preds, axis=2)
out_label_ids = np.concatenate([x["target"] for x in outputs], axis=0)
label_map = {i: label for i, label in enumerate(self.labels)}
label_map = dict(enumerate(self.labels))
out_label_list = [[] for _ in range(out_label_ids.shape[0])]
preds_list = [[] for _ in range(out_label_ids.shape[0])]
@@ -140,7 +140,7 @@ class NERTransformer(BaseTransformer):
"f1": f1_score(out_label_list, preds_list),
}
ret = {k: v for k, v in results.items()}
ret = dict(results.items())
ret["log"] = results
return ret, preds_list, out_label_list