Skip to content

Commit 4ba0cd3

Browse files
Volodymyr OrlovVolodymyr Orlov
authored andcommitted
fix: removes unnecessary print statements, fixes broken test
1 parent 01f536d commit 4ba0cd3

File tree

2 files changed

+0
-6
lines changed

2 files changed

+0
-6
lines changed

src/ensemble/random_forest_regressor.rs

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -208,10 +208,6 @@ mod tests {
208208
114.2, 115.7, 116.9,
209209
];
210210

211-
let expected_y: Vec<f64> = vec![
212-
85., 88., 88., 89., 97., 98., 99., 99., 102., 104., 109., 110., 113., 114., 115., 116.,
213-
];
214-
215211
let y_hat = RandomForestRegressor::fit(
216212
&x,
217213
&y,

src/neighbors/knn_regressor.rs

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -143,7 +143,6 @@ impl<T: RealNumber, D: Distance<Vec<T>, T>> KNNRegressor<T, D> {
143143

144144
fn predict_for_row(&self, x: Vec<T>) -> T {
145145
let search_result = self.knn_algorithm.find(&x, self.k);
146-
println!("{:?}", search_result);
147146
let mut result = T::zero();
148147

149148
let weights = self
@@ -196,7 +195,6 @@ mod tests {
196195
let y_exp = vec![2., 2., 3., 4., 4.];
197196
let knn = KNNRegressor::fit(&x, &y, Distances::euclidian(), Default::default());
198197
let y_hat = knn.predict(&x);
199-
println!("{:?}", y_hat);
200198
assert_eq!(5, Vec::len(&y_hat));
201199
for i in 0..y_hat.len() {
202200
assert!((y_hat[i] - y_exp[i]).abs() < 1e-7);

0 commit comments

Comments
 (0)