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-20 01:29:46 +09:00
parent 31394e7694
commit b6dedc9275
2 changed files with 57 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
Simulation.py
View File

@@ -152,11 +152,16 @@ class Simulation (HTS):
else:
if method == "ml":
LAST_DATA = self.stock2Vector.getLastData(stock_code, today, n=3)
data = self.stock2Vector.getRealTime(stock_code, today, LAST_DATA)
X, Y = self.stock2Vector.getVectorData(data)
result = self.stock2Vector.getRealTime(stock_code, today, LAST_DATA)
X, Y = self.stock2Vector.getVectorData(result)
predY = self.stockPredictor.predict(X, Y)
bsLine = None
predY = np.argmax(predY, axis=1)
# 이동평균, RSI, MACD, 일목균형, 볼린저밴드 상/하단을 계산한다.
data = self.buySellChecker.analyze(result)
# 사야 할 시점과 팔아야 할 시점을 체크한다.
bsLine, data = self.buySellChecker.checkTransactionML(data, stock_code, predY, isRealTime=False)
else:
LAST_DATA = self.stock2Vector.getLastData(stock_code, today)
result = self.stock2Vector.getRealTime(stock_code, today, LAST_DATA)