Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- trait L2Regularization extends com.thoughtworks.deeplearning.plugins.INDArrayWeights {
- import org.nd4j.linalg.api.ndarray.INDArray
- val l2Regularization: Double
- trait INDArrayOptimizerApi extends super.INDArrayOptimizerApi {
- this: INDArrayOptimizer =>
- abstract override def delta: INDArray = {
- import org.nd4s.Implicits._
- super.delta + weight.data * l2Regularization
- }
- }
- override type INDArrayOptimizer <: INDArrayOptimizerApi with Optimizer
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement