Skip to content

Commit 40d2b60

Browse files
committed
cargo fmt
1 parent 5dcd7b6 commit 40d2b60

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

examples/regression.rs

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -31,8 +31,7 @@ fn main() -> std::io::Result<()> {
3131
load_file("lightgbm3-sys/lightgbm/examples/regression/regression.test");
3232
assert_eq!(n_features, n_features_test);
3333

34-
let train_dataset =
35-
Dataset::from_slice(&train_xs, &train_ys, n_features, true).unwrap();
34+
let train_dataset = Dataset::from_slice(&train_xs, &train_ys, n_features, true).unwrap();
3635

3736
let params = json! {
3837
{
@@ -44,12 +43,11 @@ fn main() -> std::io::Result<()> {
4443

4544
// Train a model
4645
let booster = Booster::train(train_dataset, &params).unwrap();
47-
// Predicts floating point
46+
// Predicts floating point
4847
let y_pred = booster.predict(&test_xs, n_features, true).unwrap();
4948
// Calculate regression metrics
5049
let mean = test_ys.iter().sum::<f32>() / test_ys.len() as f32;
51-
let var =
52-
test_ys.iter().map(|&y| (y - mean).powi(2)).sum::<f32>() / test_ys.len() as f32;
50+
let var = test_ys.iter().map(|&y| (y - mean).powi(2)).sum::<f32>() / test_ys.len() as f32;
5351
let var_model = zip(&test_ys, &y_pred)
5452
.map(|(&y, &y_pred)| (y - y_pred as f32).powi(2))
5553
.sum::<f32>()

0 commit comments

Comments
 (0)