Skip to main content

run_oof_regression

Function run_oof_regression 

Source
pub fn run_oof_regression(
    model: &TabFMModel,
    x_train_raw: &[Vec<Value>],
    y_train_scaled: &[f64],
    cat_mask: &[bool],
    configs: &[MemberConfig],
    p: &EnsembleParams,
    num_folds: usize,
) -> Result<Vec<Vec<f64>>>
Expand description

Regression counterpart: [member][original_train_row] scaled OOF predictions.