From 8ca13a76d699f577b9746376f878cf8d23ec59e1 Mon Sep 17 00:00:00 2001 From: Volodymyr Orlov Date: Thu, 17 Dec 2020 19:11:47 -0800 Subject: [PATCH] fix: criterion --- benches/naive_bayes.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benches/naive_bayes.rs b/benches/naive_bayes.rs index 2a4595b..ba8cb6f 100644 --- a/benches/naive_bayes.rs +++ b/benches/naive_bayes.rs @@ -6,7 +6,7 @@ use ndarray::Array2; use smartcore::linalg::naive::dense_matrix::DenseMatrix; use smartcore::linalg::BaseMatrix; use smartcore::linalg::BaseVector; -use smartcore::naive_bayes::GaussianNB; +use smartcore::naive_bayes::gaussian::GaussianNB; pub fn gaussian_naive_bayes_fit_benchmark(c: &mut Criterion) { let mut group = c.benchmark_group("GaussianNB::fit");