Skip to main content

run_classification

zsfm_tabfm::ensemble::orchestrate

Function run_classification 

Source 
pub fn run_classification(
    model: &TabFMModel,
    x_train_raw: &[Vec<Value>],
    y_train_raw: &[Value],
    x_test_raw: &[Vec<Value>],
    cat_mask: &[bool],
    p: &EnsembleParams,
) -> Result<ClassificationOutput>