r/MachineLearning · · 1 min read

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%

https://preview.redd.it/uw26u1uo0w0h1.jpg?width=720&format=pjpg&auto=webp&s=1784e1d7d310a26fa67efc63fa5191f45433a695

https://preview.redd.it/o0ahkq7p0w0h1.jpg?width=720&format=pjpg&auto=webp&s=dcb1053c34931d11b82831c6ad8cd4755ebc5816

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!

submitted by /u/Yeet132416
[link] [comments]

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.

More from r/MachineLearning