Struct FeatureExtractor 

pub struct FeatureExtractor { /* private fields */ }

Feature extractor for logical expressions

Automatically converts TLExpr into numerical feature vectors suitable for kernel computation.

Example

use tensorlogic_sklears_kernels::{FeatureExtractor, FeatureExtractionConfig};
use tensorlogic_ir::TLExpr;

let config = FeatureExtractionConfig::new()
    .with_max_depth(3)
    .with_encode_structure(true);

let extractor = FeatureExtractor::new(config);

let expr = TLExpr::and(
    TLExpr::pred("tall", vec![]),
    TLExpr::pred("smart", vec![]),
);

let features = extractor.extract(&expr).unwrap();
println!("Feature vector: {:?}", features);

Implementations

impl FeatureExtractor

pub fn new(config: FeatureExtractionConfig) -> Self

Create a new feature extractor

pub fn extract(&self, expr: &TLExpr) -> Result<Vec<f64>>

Extract features from a logical expression

pub fn extract_batch(&self, exprs: &[TLExpr]) -> Result<Vec<Vec<f64>>>

Extract features from multiple expressions

pub fn build_vocabulary(&mut self, exprs: &[TLExpr])

Build vocabulary from a set of expressions

pub fn vocab_size(&self) -> usize

Get vocabulary size

pub fn vocabulary(&self) -> &HashMap<String, usize>

Get vocabulary

Trait Implementations

impl Clone for FeatureExtractor

fn clone(&self) -> FeatureExtractor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for FeatureExtractor

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

Formats the value using the given formatter.