/ml-functions

Some numpy + scipy style machine learning operations for ndarray

Primary LanguageRustApache License 2.0Apache-2.0

ml-functions

These are basic machine learning operations in the style of numpy and scipy that are useful for preprocessing and postprocessing machine learning data.

These functions operate on ndarray ArrayBase data structures.

Operations

  • sigmoid
  • softmax along an axis
  • exp
  • max along an axis
  • argmax along an axis
  • argsort along an axis

Usage

[dependencies]
ml-functions = { git = "https://github.com/boydjohnson/ml-functions" }