Skip to main content

run_regression

zsfm_tabfm::ensemble::orchestrate

Function run_regression 

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