arrayfire / arrayfire-ml

ArrayFire's Machine Learning Library.
BSD 3-Clause "New" or "Revised" License
102 stars 23 forks source link

Add Initializers #24

Closed jramapuram closed 7 years ago

jramapuram commented 8 years ago

Currently Nodes are defined to have spreads:

            LinearNode(const int inputSize, const int outputSize,
                       float spread = 0.05,
                       const char *name="none") :

This needs to be refactored to provide an initializations class. Examples of initializations are :

  1. Le-Cun Normal
  2. Glorot Uniform
pavanky commented 8 years ago

Erm.. Googling these terms seems to produce a lot of noise. Can you link to proper documentation ?

jramapuram commented 8 years ago

Yea, sorry. Keras has some links to the papers as well as the implementations: https://github.com/fchollet/keras/blob/master/keras/initializations.py

pavanky commented 7 years ago

This can be implemented by adding a few functions in Types.hpp.

We'll need to implement the following:

The links for lecun and glorot can be found here: https://github.com/fchollet/keras/blob/master/keras/initializers.py