#machine-learning #machine #learning #matrices #linear-algebra

easy-ml

Machine learning library providing matrices, linear algebra and automatic differentiation aimed at being easy to use

16 stable releases

1.8.2 Aug 1, 2022
1.8.1 Jan 25, 2022
1.8.0 Dec 13, 2021
1.7.2 Aug 1, 2022
1.1.0 Mar 18, 2020

#23 in Machine learning

32 downloads per month
Used in optima

MPL-2.0 license

495KB
6.5K SLoC

Easy ML

A completely deterministic machine learning library providing matrices, linear algebra and automatic differentiation over generic number types aimed at being easy to use.

Usage

Add easy-ml = "1.8" 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. The implementations of everything are more or less textbook mathematical definitions, and do not feature extensive optimisation. 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.

Features

  • serde - Optional, enables serde Serialize and Deserialize implementations.

1 - If you need to freeze your rust compiler version you should specify a minor version with a tilde requirement such as easy-ml = "~1.8". 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.

Dependencies