finetype-train 0.6.58

Training infrastructure for FineType — Sense, Entity, and Model2Vec training via Candle
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23

//! FineType training infrastructure — pure Rust via Candle.
//!
//! Provides training for:
//! - **Sense model** — Cross-attention classifier for broad category + entity subtype routing
//! - **Entity classifier** — Deep Sets MLP for entity demotion gating
//! - **Data preparation** — SOTAB + profile eval → JSONL training data
//! - **Model2Vec type embeddings** — FPS algorithm for taxonomy label matching

pub mod data;
pub mod device;
pub mod entity;
pub mod model2vec_prep;
pub mod multi_branch;
pub mod seeded_rng;
pub mod sense;
pub mod sense_train;
pub mod sibling_context;
pub mod sibling_data;
pub mod sibling_train;
pub mod training;
pub mod tui;

pub use device::get_device;