AtheMathmo / rusty-machine

Machine Learning library for Rust
https://crates.io/crates/rusty-machine/
MIT License
1.25k stars 152 forks source link

Implementing SGD for linear regression #204

Open infinite-Joy9l opened 5 years ago

infinite-Joy9l commented 5 years ago

I can see that Stochastic Gradient Descent has already been implemented. But linear regression works using simple gradient descent. What are the challenges to implementing SGD for Linear Regression.

SGD implementation: https://github.com/AtheMathmo/rusty-machine/blob/master/src/learning/optim/grad_desc.rs#L159

https://github.com/AtheMathmo/rusty-machine/blob/master/src/learning/lin_reg.rs#L157

zackmdavis commented 5 years ago

Thanks for the report! Note that unfortunately, this project is kind of suffering from a maintenance bottleneck at this point: see the comments on #181 (and please let us know in the discussion there if you want to help bring the project to life again!)