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