Waikato / wekaDeeplearning4j

Weka package for the Deeplearning4j java library
https://deeplearning.cms.waikato.ac.nz/
GNU General Public License v3.0
185 stars 202 forks source link

Use new IUpdater API #12

Closed braun-steven closed 6 years ago

braun-steven commented 7 years ago

Dl4j v0.9.0 introduced a new way to set update parameter with

NeuralNetConfiguration.Builder().updater(new Sgd(0.1))

instead of

NeuralNetConfiguration.Builder().updater(UPDATER.SGD).learningRate(0.1)

The weka wrapper have already been written in the updater package.

Due to a bug in dl4j v0.9.1 the new API cannot be used yet. Maybe we need to wait until v0.9.2.

TODO: