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.3.tar.gz
(9.5 kB
view details)
Built Distribution
Kheiron-1.0.3-py3-none-any.whl
(12.1 kB
view details)
File details
Details for the file Kheiron-1.0.3.tar.gz
.
File metadata
- Download URL: Kheiron-1.0.3.tar.gz
- Upload date:
- Size: 9.5 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.7.12
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 6893a23ccbeaa963413bb02066e7b27e7f88c06acb32ba4be2182bff1dcff89d |
|
MD5 | f7404fe5b1d9c96d9d26bde7dc87df96 |
|
BLAKE2b-256 | 91020509c8199e8b1e69129729a13287cfbd67f2c1123f08c820c024aa33ade8 |
File details
Details for the file Kheiron-1.0.3-py3-none-any.whl
.
File metadata
- Download URL: Kheiron-1.0.3-py3-none-any.whl
- Upload date:
- Size: 12.1 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/4.0.1 CPython/3.7.12
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 28736f26fde010de441e622f2db6a5e2cd9ebe010d3d0537bdd76db285ae6c26 |
|
MD5 | 242f52db76fdb79b7166702c7049254c |
|
BLAKE2b-256 | 9e94a5387f213e497ba3911bcd4dde4ebb4b6384c0a558d7cb4d66d0a3d24aae |