prefrontal 0.1.0

A blazing fast text classifier for real-time agent routing, built in Rust
Documentation 
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
26
27
28
29

use std::collections::HashMap;
use std::sync::Arc;

mod error;
mod embedding;
mod model;
pub mod builder;
mod utils;

pub use error::ClassifierError;
pub use model::Classifier;
pub use builder::{ClassifierBuilder, ClassDefinition};

/// Information about the current state and configuration of a classifier
#[derive(Debug, Clone)]
pub struct ClassifierInfo {
    /// Path to the ONNX model file
    pub model_path: String,
    /// Path to the tokenizer file
    pub tokenizer_path: String,
    /// Number of classes the classifier is trained on
    pub num_classes: usize,
    /// Labels of the classes
    pub class_labels: Vec<String>,
    /// Descriptions of the classes, wrapped in Arc for efficient cloning
    pub class_descriptions: Arc<HashMap<String, String>>,
    /// Size of the embedding vectors
    pub embedding_size: usize,
}