• Home
• org.deeplearning4j
• scalnet_2.10
• 0.6.0
• source code
• optimizer.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

org.deeplearning4j.scalnet.optimizers.optimizer.scala Maven / Gradle / Ivy

package org.deeplearning4j.scalnet.optimizers

import org.deeplearning4j.nn.api.OptimizationAlgorithm

/**
  * Optimizers for neural nets.
  *
  * @author David Kale
  */
sealed class Optimizer(val optimizationAlgorithm: OptimizationAlgorithm, val lr: Double = 1e-1)
case class SGD(override val lr: Double = 1e-1, val momentum: Double = Double.NaN, val nesterov: Boolean = false)
  extends Optimizer(optimizationAlgorithm = OptimizationAlgorithm.STOCHASTIC_GRADIENT_DESCENT)