Conditions | 4 |
Total Lines | 8 |
Code Lines | 8 |
Lines | 8 |
Ratio | 100 % |
Tests | 1 |
CRAP Score | 14.7187 |
Changes | 0 |
1 | 1 | import os |
|
19 | 1 | def save_model(self, save_path: str): |
|
20 | os.makedirs(f'model_dir/{save_path}', exist_ok=True) |
||
21 | saving_prep = lambda string: f'model_dir/{save_path}/{string}' |
||
22 | self.emb.save(saving_prep('emb.pkl')) |
||
23 | with open(saving_prep('scaler.pkl'), 'wb') as f: |
||
24 | pickle.dump(self.scaler, f, pickle.HIGHEST_PROTOCOL) |
||
25 | with open(saving_prep('model.pkl'), 'wb') as f: |
||
26 | pickle.dump(self.clf, f, pickle.HIGHEST_PROTOCOL) |
||
27 |