dsyoon/DeepStock
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

init

Browse Source
This commit is contained in:
dsyoon
2022-08-07 00:36:18 +09:00
parent f43688f0da
commit c173a6d7dc
10 changed files with 427 additions and 203 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
Simulation.py
View File

@@ -6,22 +6,24 @@ import os
from hts.HTS import HTS
from stock.util.Stock2Vector import Stock2Vector
from stock.util.StockPredictor import StockPredictor
from stock.util.LabelMaker import LabelMaker
from stock.util.StockPredictor import StockPredictor
from hts.BuySellChecker import BuySellChecker
class Simulation (HTS):
stock2Vector = None
buySellChecker = None
stockPredictor = None
def __init__(self, RESOURCE_PATH):
super().__init__(RESOURCE_PATH)
self.RESOURCE_PATH = RESOURCE_PATH
self.stock2Vector = Stock2Vector(RESOURCE_PATH)
self.labelMaker = LabelMaker(RESOURCE_PATH)
self.stockPredictor = StockPredictor()
self.buySellChecker = BuySellChecker()
self.RESOURCE_PATH = RESOURCE_PATH
self.stockPredictor = StockPredictor(RESOURCE_PATH)
#self.connect()
return
@@ -134,14 +136,15 @@ class Simulation (HTS):
def simulate(self, stock_code, today, method="rule"):
if method == "answer":
bsLine, data = self.labelMaker.makeCandidate(stock_code, today, view=True)
self.labelMaker.makeCandidate(stock_code, today, view=True)
else:
if method == "ml":
LAST_DATA = self.stock2Vector.getLastData(stock_code, today, n=10)
result = self.stock2Vector.getRealTime(stock_code, today, LAST_DATA)
LAST_DATA = self.stock2Vector.getLastData(stock_code, today, n=1)
data = self.stock2Vector.getRealTime(stock_code, today, LAST_DATA)
X, Y = self.stock2Vector.getDataset2D(data)
df, minmax_df = self.stock2Vector.preprocessData(result)
bsLine, data = self.stockPredictor.predict(df, minmax_df, isRealTime=False)
predY = self.stockPredictor.predict(X, Y)
print (predY)
else:
LAST_DATA = self.stock2Vector.getLastData(stock_code, today)
result = self.stock2Vector.getRealTime(stock_code, today, LAST_DATA)
@@ -159,17 +162,17 @@ class Simulation (HTS):
if __name__ == "__main__":
PROJECT_HOME = os.path.join(os.path.dirname(__file__))
PROJECT_HOME = os.getcwd()
RESOURCE_PATH = os.path.join(PROJECT_HOME, "resources")
# to check bying
stock_codes = {
# 252670
# 122630
"252670": ['20220805'],
"252670": ['20200731'],
}
method = "" # "ml", "answer"
method = "ml" # "ml", "answer"
for stock_code in stock_codes:
simulation = Simulation(RESOURCE_PATH)