libgrammstein 0.1.0

Hybrid language model (N-gram + Embeddings) for WFST text correction
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25

//! CLI command implementations.

pub mod convert;
pub mod corpus;
pub mod eval;
pub mod models;
pub mod query;
pub mod repl;
pub mod train;

use crate::cli::args::{Cli, Commands};
use crate::cli::error::CliResult;

/// Dispatch and run the appropriate command.
pub fn run(cli: Cli) -> CliResult<()> {
    match cli.command {
        Commands::Train(cmd) => train::run(cmd, cli.verbose, cli.quiet),
        Commands::Eval(cmd) => eval::run(cmd, cli.verbose, cli.quiet),
        Commands::Query(cmd) => query::run(cmd, cli.verbose),
        Commands::Models(cmd) => models::run(cmd, cli.verbose),
        Commands::Corpus(cmd) => corpus::run(cmd, cli.verbose),
        Commands::Convert(cmd) => convert::run(cmd, cli.verbose),
        Commands::Repl(args) => repl::run(args),
    }
}