def let_train_invest_twins(self, corp_code, corp_name, no): """겨별 세션과 통합세션에서 예측한 값의 평균을 예측값으로 한다.""" stocks = Stocks(self.params) trains_data = TrainsData(self.params) learning = Learning(self.params) params_all = TrainParams('ALL_CORPS') learning_all = Learning(params_all) invest = MockInvestment(self.params) stock_data = stocks.get_stock_data(corp_code) data_params, scaler_close, dataX_last = trains_data.get_train_test( stock_data) rmse_val, train_cnt, rmse_vals, test_predict = learning.let_learning( corp_code, data_params) rmse_val_all, train_cnt_all, rmse_vals_all, test_predict_all = learning_all.let_learning( corp_code, data_params) last_money, last_predict, invest_predicts, all_invest_money = \ invest.let_invest_and_all(corp_code, dataX_last, data_params, params_all) rmse_val = np.mean([rmse_val, rmse_val_all]) train_cnt = train_cnt + train_cnt_all print(no, corp_code, corp_name, rmse_val, last_money, all_invest_money, train_cnt) return [ no, corp_code, corp_name, rmse_val, last_money, all_invest_money, train_cnt ]
def let_train_only(self, corp_code, stock_data, invest_count=None): """입력한 회사에 대해서 학습시킨다""" trains_data = TrainsData(self.params) learning = Learning(self.params) data_params, scaler_close, dataX_last = trains_data.get_train_test( stock_data, invest_count) rmse_val, train_cnt, rmse_vals, test_predict = learning.let_learning( corp_code, data_params) return rmse_val, train_cnt, data_params, dataX_last, scaler_close