search

jinlow / forust

A lightweight gradient boosted decision tree package.
https://jinlow.github.io/forust/
Apache License 2.0
56 stars 6 forks source link

Regression? #65

Closed ronniec95 closed 10 months ago

ronniec95 commented 10 months ago

Is there a way to get regression values?

jinlow commented 10 months ago

Regression is supported using “SquaredLoss“ via the objective_type.
in python…

from forust import GradientBooster
model = GradientBooster(objective_type="LogLoss")

In rust…


use forust_ml::GradientBooster;
use forust_ml::objective::ObjectiveType;
let model = GradientBooster::new().set_objective_type(ObjectiveType::SquaredLoss);
ronniec95 commented 10 months ago

Thank you; this worked great