Skip to content

Commit

Permalink
cargo fmt
Browse files Browse the repository at this point in the history
  • Loading branch information
Mottl committed Jun 24, 2023
1 parent 5dcd7b6 commit 40d2b60
Showing 1 changed file with 3 additions and 5 deletions.
8 changes: 3 additions & 5 deletions examples/regression.rs
Original file line number Diff line number Diff line change
Expand Up @@ -31,8 +31,7 @@ fn main() -> std::io::Result<()> {
load_file("lightgbm3-sys/lightgbm/examples/regression/regression.test");
assert_eq!(n_features, n_features_test);

let train_dataset =
Dataset::from_slice(&train_xs, &train_ys, n_features, true).unwrap();
let train_dataset = Dataset::from_slice(&train_xs, &train_ys, n_features, true).unwrap();

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

// Train a model
let booster = Booster::train(train_dataset, &params).unwrap();
// Predicts floating point
// Predicts floating point
let y_pred = booster.predict(&test_xs, n_features, true).unwrap();
// Calculate regression metrics
let mean = test_ys.iter().sum::<f32>() / test_ys.len() as f32;
let var =
test_ys.iter().map(|&y| (y - mean).powi(2)).sum::<f32>() / test_ys.len() as f32;
let var = test_ys.iter().map(|&y| (y - mean).powi(2)).sum::<f32>() / test_ys.len() as f32;
let var_model = zip(&test_ys, &y_pred)
.map(|(&y, &y_pred)| (y - y_pred as f32).powi(2))
.sum::<f32>()
Expand Down

0 comments on commit 40d2b60

Please sign in to comment.