Struct SmallTsetlinMachine 

pub struct SmallTsetlinMachine<const N: usize, const C: usize> { /* private fields */ }

Binary classification Tsetlin Machine with compile-time known dimensions.

Stack-allocated with zero heap allocations. All loops unrolled at compile time.

Type Parameters

• N - Number of features (compile-time constant)
• C - Number of clauses (compile-time constant, must be even)

Performance

Up to 3x faster than dynamic TsetlinMachine for small dimensions due to:

• No heap allocations
• Full loop unrolling
• Better cache locality

Example

use tsetlin_rs::SmallTsetlinMachine;

// XOR with 2 features and 20 clauses
let mut tm: SmallTsetlinMachine<2, 20> = SmallTsetlinMachine::new(100, 15);

let x = [[0, 0], [0, 1], [1, 0], [1, 1]];
let y = [0u8, 1, 1, 0];

tm.fit(&x, &y, 200, 42);
assert!(tm.evaluate(&x, &y) >= 0.75);

Implementations

impl<const N: usize, const C: usize> SmallTsetlinMachine<N, C>

pub fn new(n_states: i16, threshold: i32) -> Self

Creates new machine with given states and threshold.

Half clauses get +1 polarity, half get -1.

Panics

Debug-asserts that C is even.

pub fn with_s(n_states: i16, threshold: i32, s: f32) -> Self

Creates machine with custom specificity parameter.

pub const fn n_features(&self) -> usize

Returns the number of features (compile-time constant).

pub const fn n_clauses(&self) -> usize

Returns the number of clauses (compile-time constant).

pub fn threshold(&self) -> f32

Returns current threshold.

pub const fn clauses(&self) -> &[SmallClause<N>; C]

Returns read-only access to clauses.

pub fn sum_votes(&self, x: &[u8; N]) -> i32

Sum of clause votes for input x.

pub fn predict(&self, x: &[u8; N]) -> u8

Predicts class (0 or 1).

pub fn train_one(&mut self, x: &[u8; N], y: u8, rng: &mut impl Rng)

Trains on single example.

pub fn fit(&mut self, x: &[[u8; N]], y: &[u8], epochs: usize, seed: u64)

Simple training for given epochs.

pub fn evaluate(&self, x: &[[u8; N]], y: &[u8]) -> f32

Evaluates accuracy on test data.

Trait Implementations

impl<const N: usize, const C: usize> Clone for SmallTsetlinMachine<N, C>

fn clone(&self) -> SmallTsetlinMachine<N, C>

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<const N: usize, const C: usize> Debug for SmallTsetlinMachine<N, C>

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.