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

Fix suggested by @bhadreshpsavani (#11660)

Browse Source
This commit is contained in:
Matt
2021-05-10 14:28:04 +01:00
committed by GitHub
parent 575c979144
commit ef8d32c5ea
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
examples/tensorflow/text-classification/run_text_classification.py
View File

@@ -522,7 +522,7 @@ def main():
# region Prediction losses # region Prediction losses
# This section is outside the scope() because it's very quick to compute, but behaves badly inside it # This section is outside the scope() because it's very quick to compute, but behaves badly inside it
if "label" in datasets["test"].features: if "test" in datasets and "label" in datasets["test"].features:
print("Computing prediction loss on test labels...") print("Computing prediction loss on test labels...")
labels = datasets["test"]["label"] labels = datasets["test"]["label"]
loss = float(loss_fn(labels, predictions).numpy()) loss = float(loss_fn(labels, predictions).numpy())