(self,
config
)
| 307 | |
| 308 | class Run(): |
| 309 | def __init__(self, |
| 310 | config |
| 311 | ): |
| 312 | |
| 313 | self.model_name = config['model_name'] |
| 314 | self.mode_eval = config['mode_eval'] |
| 315 | self.fold = config['fold'] |
| 316 | self.data_type = 'SVFEND' |
| 317 | |
| 318 | self.epoches = config['epoches'] |
| 319 | self.batch_size = config['batch_size'] |
| 320 | self.num_workers = config['num_workers'] |
| 321 | self.epoch_stop = config['epoch_stop'] |
| 322 | self.seed = config['seed'] |
| 323 | self.device = config['device'] |
| 324 | self.lr = config['lr'] |
| 325 | self.lambd=config['lambd'] |
| 326 | self.save_param_dir = config['path_param'] |
| 327 | self.path_tensorboard = config['path_tensorboard'] |
| 328 | self.dropout = config['dropout'] |
| 329 | self.weight_decay = config['weight_decay'] |
| 330 | self.event_num = 616 |
| 331 | self.mode ='normal' |
| 332 | |
| 333 | |
| 334 | def get_dataloader(self,data_type,data_fold): |
nothing calls this directly
no outgoing calls
no test coverage detected