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 |