Function sb_fit_predict_regressor

crates/pyopenquant/src/sb_bagging.rs:56–82

(
    py: Python<'_>,
    x: Vec<Vec<f64>>,
    y: Vec<f64>,
    ind_mat: Vec<Vec<u8>>,
    n_estimators: usize,
    max_samples: f64,
    max_features: f64,
    random_state: u64,
    sample_weight: 

    random_state=42,
    sample_weight=None
))]
fn sb_fit_predict_regressor(
    py: Python<'_>,
    x: Vec<Vec<f64>>,
    y: Vec<f64>,
    ind_mat: Vec<Vec<u8>>,
    n_estimators: usize,
    max_samples: f64,
    max_features: f64,
    random_state: u64,
    sample_weight: Option<Vec<f64>>,
) -> PyResult<PyObject> {
    let x_mat = matrix_from_rows(x)?;

    let mut reg =
        openquant::sb_bagging::SequentiallyBootstrappedBaggingRegressor::new(random_state);
    reg.n_estimators = n_estimators;
    reg.max_samples = openquant::sb_bagging::MaxSamples::Float(max_samples);
    reg.max_features = openquant::sb_bagging::MaxFeatures::Float(max_features);
    reg.oob_score = true;

    reg.fit(&x_mat, &y, &ind_mat, sample_weight.as_deref()).map_err(to_py_err)?;
    let predictions = reg.predict(&x_mat).map_err(to_py_err)?;

    let d = PyDict::new(py);
    d.set_item("predictions", predictions)?;
    d.set_item("oob_score", reg.oob_score_value)?;
    Ok(d.into_pyobject(py).unwrap().into_any().unbind())
}