Built Support Vector Machine(SVM) from scratch in Rust [P]
Mirrored from r/MachineLearning for archival readability. Support the source by reading on the original site.
Built my own SVM classifier from scratch in Rust. It uses SMO optimization, have linear and rbf kernel, uses grid search to tune the hyperparameters.
I tested it on two datasets one using Linear dataset and other using RBF, these were the results:
| Dataset | Kernel | Accuracy | Recall | F1 |
|---|---|---|---|---|
| Banknote Auth | Linear | 96% | 94% | 95% |
| Breast Cancer | RBF | 93% | 100% | 92% |
The plot.rs file, used for plotting only was written using AI as I could not wrap my head around plotters crate, apart from that everything was by my own.
Repo Link: Github Repo
Happy to get some feedback!
[link] [comments]
More from r/MachineLearning
-
What kinds of models are people training with document data? [P]
May 13
-
Have the "on-hold" durations been getting longer for arXiv submissions? [D]
May 13
-
Image generation models running locally on limited resources [P]
May 13
-
EEML Summer School (Eastern European ML) - Anyone here got accepted? [D]
May 13
Discussion (0)
Sign in to join the discussion. Free account, 30 seconds — email code or GitHub.
Sign in →No comments yet. Sign in and be the first to say something.