/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.

Primary LanguageRustMIT LicenseMIT

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