Exemplo n.º 1
0
 def load_model(self, feature):
     model_path = os.path.join(f'../model/model/{feature}',
                               f'{self.run_fold_name}.h5')
     scaler_path = os.path.join(f'../model/model/{feature}',
                                f'{self.run_fold_name}-scaler.pkl')
     self.model = load_model(model_path)
     self.scaler = Util.load(scaler_path)
Exemplo n.º 2
0
 def load_model(self, feature):
     model_path = os.path.join(f'../model/model/{feature}',
                               f'{self.run_fold_name}.model')
     self.model = Util.load(model_path)