Skip to main content

run_members_regression

Function run_members_regression 

Source
pub fn run_members_regression(
    model: &TabFMModel,
    x_train_raw: &[Vec<Value>],
    y_train_scaled: &[f64],
    x_query_raw: &[Vec<Value>],
    cat_mask: &[bool],
    configs: &[MemberConfig],
    outlier_threshold: f64,
    batch_size: Option<usize>,
) -> Result<Vec<Vec<f64>>>
Expand description

Same idea for regression: returns [member][query_row] scaled (not yet inverse-transformed) predictions.