nkaush / naive-bayes-rs

A Rust library with homemade machine learning models to classify the MNIST dataset. Built in an attempt to get familiar with advanced Rust concepts.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

mnist-classifier

Ideas

  • UPDATED: Finish CLI Flags
  • Parallelize conputationally intensive functions
  • Class-based naive bayes
  • README
  • Image parsing
  • Confusion matrix saving
  • Port model code to separate module
  • examples of custom models -> use STAT200 datasets for classification
  • remove panics and port to custom error codes
  • make test return confusion matrix and accuracy

About

A Rust library with homemade machine learning models to classify the MNIST dataset. Built in an attempt to get familiar with advanced Rust concepts.

License:MIT License


Languages

Language:Rust 99.8%Language:Shell 0.2%