6 stable releases

1.4.0 Sep 20, 2020
1.3.1 Jun 9, 2020
1.2.0 Apr 25, 2020
1.1.0 Mar 18, 2020
1.0.0 Feb 9, 2020

#30 in Machine learning

Download history 35/week @ 2020-06-30 49/week @ 2020-07-07 63/week @ 2020-07-14 8/week @ 2020-07-21 2/week @ 2020-07-28 26/week @ 2020-08-04 16/week @ 2020-08-11 47/week @ 2020-08-18 11/week @ 2020-08-25 7/week @ 2020-09-01 44/week @ 2020-09-08 85/week @ 2020-09-15 14/week @ 2020-09-22 33/week @ 2020-09-29 13/week @ 2020-10-06 13/week @ 2020-10-13

138 downloads per month

MPL-2.0 license

310KB
3.5K SLoC

Easy ML

A completely deterministic matrix and linear algebra library over generic number types aimed at being easy to use.

Usage

Add easy-ml = "1" to your [dependencies]1.

Overview

This is a pure Rust library which makes heavy use of passing closures, iterators, generic types, and other rust idioms that machine learning libraries which wrap around another language backend could never provide easily. This library tries to provide adequate documentation to explain what the functions compute, what the computations mean, and examples of tasks you might want to do with this library:

  • Linear Regression
  • k-means Clustering
  • Logistic Regression
  • Naïve Bayes
  • Feedforward neural networks
  • Backprop with Automatic Differentiation
  • using a custom numeric type such as num_bigint::BigInt
  • Handwritten digit recognition in the browser

This library is not designed for deep learning. I have not tried to optimise much of anything. The implementations of everything are more or less textbook mathematical definitions. You might find that you need to use a faster library once you've explored your problem, or that you need something that's not implemented here and have to switch. I hope that being able to at least start here may be of use.

State of library

This library is currently usable for simple linear algebra tasks like linear regression and for storing 2 dimensional data. This library also has support for forward and reverse automatic differentiation and can train simple feedforward neural networks. Currently there is no support for 3d or higher dimensional data, though matrices can be nested into each other. This library supports compilation to Web Assembly.

Level of abstraction

Where as other machine learning libraries often create objects/structs to represent algorithms like linear regression or k-means, Easy ML instead only represents the data in structs and the consuming code determines all of the control flow. While this may take more effort to write for consuming code initially, it means making changes to the algorithms are much easier.


1 - If you need to freeze your rust compiler version you should specify a minor version such as easy-ml = "1.4". Easy ML will not introduce breaking API changes between minor versions, but does follow the latest stable version of rust, and thus may introduce dependencies on newer language features (eg const generics) in minor version updates.

No runtime deps