Implement predict_proba for DecisionTreeClassifier (#287)

* Implement predict_proba for DecisionTreeClassifier
* Some automated fixes suggested by cargo clippy --fix
This commit is contained in:
Lorenzo
2025-01-20 18:50:00 +00:00
committed by GitHub
parent 4523ac73ff
commit 3da433f757
13 changed files with 166 additions and 61 deletions
+3 -1
View File
@@ -212,7 +212,9 @@ mod tests_fastpair {
use crate::linalg::basic::{arrays::Array, matrix::DenseMatrix};
/// Brute force algorithm, used only for comparison and testing
pub fn closest_pair_brute(fastpair: &FastPair<f64, DenseMatrix<f64>>) -> PairwiseDistance<f64> {
pub fn closest_pair_brute(
fastpair: &FastPair<'_, f64, DenseMatrix<f64>>,
) -> PairwiseDistance<f64> {
use itertools::Itertools;
let m = fastpair.samples.shape().0;