linfa
linfa copied to clipboard
rust-ml
A Rust machine learning framework.
Following #183 tips. This is an implementation of Bernoulli Naive Bayes classifier derived from Multinomial NB.
We should add Python bindings to the public API of `linfa` crates. This will allow us to fairly benchmark `linfa` against `scikit-learn`, which also uses a Python API, as well...
Now, there is sigmoid function implemented in both `linfa-logistic` and `linfa-ftrl` crates. There can be list of all function needed to be refactored within all crates - [ ] sigmoid...
Trying to compile `linfa` master, I stumbled on the following: ``` > cargo test --all --release [...] Compiling mnist v0.5.0 Compiling time v0.2.26 error[E0433]: failed to resolve: could not find...
The goal of this PR is to add multi-task ElasticNet to the `elasticnet` crate. A quick roadmap: - [x] Write block coordinate descent - [x] Write dual gap for multi-task...
`winequality` isn't a good dataset for `linfa-bayes`, so we need to replace it with something better in the example code. The 20 Newsgroups is already used [here](https://github.com/rust-ml/linfa/blob/master/algorithms/linfa-preprocessing/examples/count_vectorization.rs). We just need...
I can certainly implement MultiTaskLasso, as well as GroupLasso, in the ElasticNet package. Would you consider a valuable addition to the crate?
In high dimensional statistics, we usually rely on CSC and CSR arrays when matrices are very sparse. It is particularly for speeding up the coordinate descent procedure. I can implement...
One suggestion to prevent bringing in more dependencies than a user needs would be to make serde an optional dependency and adjust the `Serialize` and `Deserialize` derives be like `#[cfg_attr(feature...
