函数作用:模型初始化 FC1 -> Sigmoid -> FC2 -> Softmax
(self)
| 292 | self.is_initialized = False |
| 293 | |
| 294 | def _set_params(self): |
| 295 | """ |
| 296 | 函数作用:模型初始化 |
| 297 | FC1 -> Sigmoid -> FC2 -> Softmax |
| 298 | """ |
| 299 | self.layers = OrderedDict() |
| 300 | self.layers["FC1"] = FullyConnected( |
| 301 | n_out=self.hidden_dims_1, |
| 302 | acti_fn="sigmoid", |
| 303 | init_w=self.init_w, |
| 304 | optimizer=self.optimizer |
| 305 | ) |
| 306 | self.layers["FC2"] = FullyConnected( |
| 307 | n_out=self.hidden_dims_2, |
| 308 | acti_fn="affine(slope=1, intercept=0)", |
| 309 | init_w=self.init_w, |
| 310 | optimizer=self.optimizer |
| 311 | ) |
| 312 | self.is_initialized = True |
| 313 | |
| 314 | def forward(self, X_train): |
| 315 | Xs = {} |