| Conditions | 5 |
| Total Lines | 19 |
| Code Lines | 15 |
| Lines | 0 |
| Ratio | 0 % |
| Changes | 0 | ||
| 1 | # Author: Simon Blanke |
||
| 16 | def train_model(self, para_dict): |
||
| 17 | start_time = time.time() |
||
| 18 | results = self.func_(para_dict, self.X, self.y) |
||
| 19 | eval_time = time.time() - start_time |
||
| 20 | |||
| 21 | if isinstance(results, tuple): |
||
| 22 | score = results[0] |
||
| 23 | model = results[1] |
||
| 24 | elif ( |
||
| 25 | isinstance(results, float) |
||
| 26 | or isinstance(results, np.float64) |
||
| 27 | or isinstance(results, np.float32) |
||
| 28 | ): |
||
| 29 | score = results |
||
| 30 | model = None |
||
| 31 | else: |
||
| 32 | print("Error: model function must return float or tuple") |
||
| 33 | |||
| 34 | return score, eval_time, model |
||
| 35 |