Skip to content

GitLab

Unverified Commit ce6add8e authored by Nathan Raw's avatar Nathan Raw Committed by GitHub
Browse files

🐛 fix small model card bugs (#13310) 

* 🐛 fix small model card bugs

* 💄 style
parent 139e8301
Hide whitespace changes
Inline Side-by-side
Showing with 9 additions and 6 deletions
src/transformers/modelcard.py
View file @ ce6add8e
...@@ -426,12 +426,15 @@ class TrainingSummary: ...@@ -426,12 +426,15 @@ class TrainingSummary:
result["dataset"]["args"] = dataset_arg_mapping[ds_tag] result["dataset"]["args"] = dataset_arg_mapping[ds_tag]
if len(metric_mapping) > 0: if len(metric_mapping) > 0:
result["metrics"] = []
for metric_tag, metric_name in metric_mapping.items(): for metric_tag, metric_name in metric_mapping.items():
result["metric"] = { result["metrics"].append(
"name": metric_name, {
"type": metric_tag, "name": metric_name,
"value": self.eval_results[metric_name], "type": metric_tag,
} "value": self.eval_results[metric_name],
}
)
model_index["results"].append(result) model_index["results"].append(result)
...@@ -446,7 +449,7 @@ class TrainingSummary: ...@@ -446,7 +449,7 @@ class TrainingSummary:
metadata = _insert_values_as_list(metadata, "tags", self.tags) metadata = _insert_values_as_list(metadata, "tags", self.tags)
metadata = _insert_values_as_list(metadata, "datasets", self.dataset_tags) metadata = _insert_values_as_list(metadata, "datasets", self.dataset_tags)
metadata = _insert_values_as_list(metadata, "metrics", list(metric_mapping.keys())) metadata = _insert_values_as_list(metadata, "metrics", list(metric_mapping.keys()))
metadata["model_index"] = self.create_model_index(metric_mapping) metadata["model-index"] = self.create_model_index(metric_mapping)
return metadata return metadata
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment