Skip to content

Instantly share code, notes, and snippets.

@charlieoneill11

charlieoneill11/preds.py

Created May 1, 2022 07:01
Show Gist options
  • Save charlieoneill11/6b3830e7ad446abe5466b21f2509e126 to your computer and use it in GitHub Desktop.
Save charlieoneill11/6b3830e7ad446abe5466b21f2509e126 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
preds.py
preds_output = trainer.predict(offensive_encoded["validation"])
preds_output.metrics
> {'test_loss': 0.42069146037101746,
'test_accuracy': 0.8013595166163142,
'test_f1': 0.8079235239615667,
'test_runtime': 3.3561,
'test_samples_per_second': 394.507,
'test_steps_per_second': 6.257}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment