Kheiron: Train and evaluate libary for Pytorch model.
Project description
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
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Kheiron-1.0.2.tar.gz
(9.7 kB
view hashes)
Built Distribution
Kheiron-1.0.2-py3-none-any.whl
(11.9 kB
view hashes)