Allow KNN with k=1

This commit is contained in:
Luis Moreno
2020-10-16 12:28:30 -04:00
parent a2588f6f45
commit 92dad01810
2 changed files with 5 additions and 2 deletions
+2 -2
View File
@@ -116,9 +116,9 @@ impl<T: RealNumber, D: Distance<Vec<T>, T>> KNNRegressor<T, D> {
)));
}
if parameters.k <= 1 {
if parameters.k < 1 {
return Err(Failed::fit(&format!(
"k should be > 1, k=[{}]",
"k should be > 0, k=[{}]",
parameters.k
)));
}