def run_predict_all(self) -> None: """学習データすべてで学習したモデルにより、テストデータの予測を行う あらかじめrun_train_allを実行しておく必要がある """ self.logger.info(f'{self.run_name} - start prediction all') test_x = self.load_x_test() # 学習データ全てで学習したモデルで予測を行う i_fold = 'all' model = self.build_model(i_fold) model.load_model(self.out_dir_name) pred = model.predict(test_x) # 予測結果の保存 Util.dump(pred, f'../model/pred/{self.run_name}-test.pkl') self.logger.info(f'{self.run_name} - end prediction all')
def save_model(self, path): model_path = os.path.join(path, f'{self.run_fold_name}.model') os.makedirs(os.path.dirname(model_path), exist_ok=True) Util.dump(self.model, model_path)
def save_model(self, path): model_path = os.path.join(path, f'{self.run_fold_name}.h5') scaler_path = os.path.join(path, f'{self.run_fold_name}-scaler.pkl') os.makedirs(os.path.dirname(model_path), exist_ok=True) self.model.save(model_path) Util.dump(self.scaler, scaler_path)