zantetsu-trainer 0.1.0

Background sync and RLAIF training loop for Zantetsu
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12

//! # Zantetsu Trainer
//!
//! Training pipeline for the neural CRF model.
//! Includes data loading, model training, and model export.

pub mod data;
pub mod model;
pub mod trainer;

pub use data::{load_bio_dataset, CharVocab, TrainingExample, BIO_LABELS};
pub use model::{viterbi_decode, CrfModel, NUM_LABELS};
pub use trainer::{run_training, Trainer};