datnnt1997 / Kheiron

Kheiron

Kheiron: Train and evaluate libary for Pytorch model.

Author: Trong-Dat Ngo.

---

Examples

Text classification

options = TrainingOptions(task='text-classification',
                          train_batch_size=8,
                          eval_batch_size=8,
                          metric_for_best_model='macro_f1',
                          greater_is_better=True,
                          track_metrics=True)

trainer = Trainer(model=model,
                  opts=options,
                  train_set=tokenized_sets['train'],
                  eval_set=tokenized_sets['test'],
                  collate_fn=data_collator)

trainer.train()

Token classification

Sammarization

Translation