Skip to content

High performance implementation of Extreme Learning Machines (fast randomized neural networks).

Notifications You must be signed in to change notification settings

zhufengGNSS/hpelm

 
 

Repository files navigation

High Performance ELM

Extreme Learning Machine (ELM) with model selection and regularizations.

In-memory ELM works, check hpelm/tests folder. MAGMA acceleration works, check hpelm/acc/setup_gpu.py.

Example usage:

>>> from hpelm import ELM
>>> elm = ELM(X.shape[1], T.shape[1])
>>> elm.add_neurons(20, "sigm")
>>> elm.add_neurons(10, "rbf_l2")
>>> elm.train(X, T, "LOO")
>>> Y = elm.predict(X)

If you use the toolbox, cite our paper "High Performance Extreme Learning Machines: A Complete Toolbox for Big Data Applications" that will be published in IEEE Access.

About

High performance implementation of Extreme Learning Machines (fast randomized neural networks).

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C 60.5%
  • Python 31.7%
  • MATLAB 3.6%
  • C++ 2.4%
  • Makefile 0.9%
  • Batchfile 0.9%