wamsy's picture
Update README.md
a160775
metadata
license: unlicense
datasets:
  - sem_eval_2018_task_1
language:
  - en
metrics:
  - accuracy
library_name: bertopic
pipeline_tag: text-classification

image/png

{'eval_loss': 0.3406790494918823, 'eval_f1': 0.707515557683102, 'eval_roc_auc': 0.8064406569750754, 'eval_accuracy': 0.2584650112866817, 'eval_runtime': 6.4135, 'eval_samples_per_second': 138.146, 'eval_steps_per_second': 17.307, 'epoch': 15.0}