(self, params)
| 80 | } |
| 81 | |
| 82 | def set_params(self, params): |
| 83 | # params is a flat list |
| 84 | # unflatten into individual weights |
| 85 | D, M, K = self.D, self.M, self.K |
| 86 | self.W1 = params[:D * M].reshape(D, M) |
| 87 | self.b1 = params[D * M:D * M + M] |
| 88 | self.W2 = params[D * M + M:D * M + M + M * K].reshape(M, K) |
| 89 | self.b2 = params[-K:] |
| 90 | |
| 91 | |
| 92 | def evolution_strategy( |
no outgoing calls
no test coverage detected