Skip to content

Commit e12d975

Browse files

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

benches/naive_bayes.rs

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
use criterion::BenchmarkId;
2-
use criterion::{criterion_group, criterion_main, Criterion};
2+
use criterion::{black_box, criterion_group, criterion_main, Criterion};
33

44
use nalgebra::DMatrix;
55
use ndarray::Array2;
@@ -25,7 +25,7 @@ pub fn gaussian_naive_bayes_fit_benchmark(c: &mut Criterion) {
2525
n_samples,
2626
|b, _| {
2727
b.iter(|| {
28-
GaussianNB::fit(&x, &y, Default::default()).unwrap();
28+
GaussianNB::fit(black_box(&x), black_box(&y), Default::default()).unwrap();
2929
})
3030
},
3131
);
@@ -43,7 +43,7 @@ pub fn gaussian_naive_matrix_datastructure(c: &mut Criterion) {
4343
let y = <DenseMatrix<f64> as BaseMatrix<f64>>::RowVector::from_array(&classes);
4444

4545
b.iter(|| {
46-
GaussianNB::fit(&x, &y, Default::default()).unwrap();
46+
GaussianNB::fit(black_box(&x), black_box(&y), Default::default()).unwrap();
4747
})
4848
});
4949

@@ -52,7 +52,7 @@ pub fn gaussian_naive_matrix_datastructure(c: &mut Criterion) {
5252
let y = <Array2<f64> as BaseMatrix<f64>>::RowVector::from_array(&classes);
5353

5454
b.iter(|| {
55-
GaussianNB::fit(&x, &y, Default::default()).unwrap();
55+
GaussianNB::fit(black_box(&x), black_box(&y), Default::default()).unwrap();
5656
})
5757
});
5858

@@ -61,7 +61,7 @@ pub fn gaussian_naive_matrix_datastructure(c: &mut Criterion) {
6161
let y = <DMatrix<f64> as BaseMatrix<f64>>::RowVector::from_array(&classes);
6262

6363
b.iter(|| {
64-
GaussianNB::fit(&x, &y, Default::default()).unwrap();
64+
GaussianNB::fit(black_box(&x), black_box(&y), Default::default()).unwrap();
6565
})
6666
});
6767
}

0 commit comments

Comments
 (0)