func optimization(rate: Double) {
   data.weight1 = data.weight1 - rate * data.gradW1
   data.bias1 = data.bias1 - rate * data.gradB1
   data.weight2 = data.weight2 - rate * data.gradW2
   data.bias2 = data.bias2 - rate * data.gradB2
}