Advertisement
Guest User

Untitled

a guest
Jun 20th, 2019
66
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.73 KB | None | 0 0
  1. private fun optimizeParameters( gradients : ArrayList<Array<Any>> , learningRate : Double ) {
  2. val weightGradientsList = ArrayList<DoubleArray>()
  3. for( gradient in gradients ) {
  4. weightGradientsList.add( gradient[0] as DoubleArray )
  5. }
  6. val weightGradients = MathOps.multidimMean( weightGradientsList.toTypedArray() ).toDoubleArray()
  7.  
  8. val biasGradientsList = ArrayList<Double>()
  9. for( gradient in gradients ) {
  10. biasGradientsList.add( gradient[1] as Double )
  11. }
  12. val biasGradients = ( biasGradientsList.toTypedArray() ).average()
  13. this.weights = MathOps.subtract( this.weights , MathOps.multiplyScalar( weightGradients , learningRate ) )
  14. this.bias = this.bias - ( biasGradients * learningRate )
  15. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement