tracel-ai / burn

Burn is a new comprehensive dynamic Deep Learning Framework built using Rust with extreme flexibility, compute efficiency and portability as its primary goals.
https://burn.dev
Apache License 2.0
7.94k stars 382 forks source link

traditional machine learning model #1673

Closed wangjiawen2013 closed 3 months ago

wangjiawen2013 commented 3 months ago

Hi, It is said that "Burn aims to build a full-fledged machine learning stack in Rust" and the following example are presented (https://medium.com/@athan.seal/candle-vs-burn-comparing-rust-machine-learning-frameworks-4dbd59c332a1):

use burn::prelude::*;

fn main() {
    let dataset = Dataset::from_csv("data.csv")?;
    let model = LogisticRegression::new(dataset.num_features());

    let optimizer = GradientDescent::new(0.01);
    let mut history = optimizer.fit(&dataset, &model, 100)?;

    println!("Training loss: {}", history.losses().last().unwrap());
}

Where can I find more examples on traditional machine learning such as logisticregression, svm, dimension reduction, naive bayes and so on ? I didn't find LogisticRegression on burn's documentation. Is the above comment on burn and candle true ?

laggui commented 3 months ago

Actually, Burn is not meant to be a "full-fledged machine learning stack". Burn is a deep learning framework.

We do not have traditional machine learning methods such as svm, naive bayes, etc.

Not sure if the linked post included the code to their example, but the closest thing we have is a simple regression example.

If you're looking for traditional ML methods in Rust, take a look at Are We Learning Yet. I just stumbled upon this recently, it could be a helpful resource for you.