"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "ae454f41d472617f3af0a184482c576c722b7e7e"
Unverified Commit eb330e89 authored by Bhadresh Savani's avatar Bhadresh Savani Committed by GitHub
Browse files

fixed typo (#10861)

parent e21f89f6
...@@ -1722,7 +1722,7 @@ class Trainer: ...@@ -1722,7 +1722,7 @@ class Trainer:
return output.metrics return output.metrics
def predict( def predict(
self, test_dataset: Dataset, ignore_keys: Optional[List[str]] = None, metric_key_prefix: str = "eval" self, test_dataset: Dataset, ignore_keys: Optional[List[str]] = None, metric_key_prefix: str = "test"
) -> PredictionOutput: ) -> PredictionOutput:
""" """
Run prediction and returns predictions and potential metrics. Run prediction and returns predictions and potential metrics.
...@@ -1737,9 +1737,9 @@ class Trainer: ...@@ -1737,9 +1737,9 @@ class Trainer:
ignore_keys (:obj:`Lst[str]`, `optional`): ignore_keys (:obj:`Lst[str]`, `optional`):
A list of keys in the output of your model (if it is a dictionary) that should be ignored when A list of keys in the output of your model (if it is a dictionary) that should be ignored when
gathering predictions. gathering predictions.
metric_key_prefix (:obj:`str`, `optional`, defaults to :obj:`"eval"`): metric_key_prefix (:obj:`str`, `optional`, defaults to :obj:`"test"`):
An optional prefix to be used as the metrics key prefix. For example the metrics "bleu" will be named An optional prefix to be used as the metrics key prefix. For example the metrics "bleu" will be named
"eval_bleu" if the prefix is "eval" (default) "test_bleu" if the prefix is "test" (default)
.. note:: .. note::
......
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