import com.thoughtworks.deeplearning.plugins.INDArrayWeights

trait INDArrayLearningRate extends INDArrayWeights {

  val learningRate: Double

  trait INDArrayOptimizerApi extends super.INDArrayOptimizerApi { this: INDArrayOptimizer =>

    import org.nd4s.Implicits._
    override def delta: INDArray = {
      super.delta * learningRate
    }
  }

  override type INDArrayOptimizer <: Optimizer with INDArrayOptimizerApi

}