Conditions | 4 |
Total Lines | 7 |
Code Lines | 7 |
Lines | 7 |
Ratio | 100 % |
Tests | 1 |
CRAP Score | 14.0742 |
Changes | 0 |
1 | 1 | import os |
|
10 | 1 | def load_model(self, load_path: str): |
|
11 | loading_prep = lambda string: f'model_dir/{load_path}/{string}' |
||
12 | self.emb.load(loading_prep('emb.pkl')) |
||
13 | with open(loading_prep('model.pkl'), 'rb') as f: |
||
14 | self.scaler = pickle.load(f) |
||
15 | with open(loading_prep('scaler.pkl'), 'rb') as f: |
||
16 | self.scaler = pickle.load(f) |
||
17 | |||
26 |