Skip to content

Commit

Permalink
fixed typo (huggingface#10861)
Browse files Browse the repository at this point in the history
  • Loading branch information
bhadreshpsavani authored and Iwontbecreative committed Jul 15, 2021
1 parent a20cfd4 commit 589ae1a
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions src/transformers/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -1722,7 +1722,7 @@ def evaluate(
return output.metrics

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:
"""
Run prediction and returns predictions and potential metrics.
Expand All @@ -1737,9 +1737,9 @@ def predict(
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
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
"eval_bleu" if the prefix is "eval" (default)
"test_bleu" if the prefix is "test" (default)
.. note::
Expand Down

0 comments on commit 589ae1a

Please sign in to comment.