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

init

Browse Source
This commit is contained in:
dosang.yoon
2022-08-30 14:37:23 +09:00
parent 42710247be
commit 8fdeba4ea5
1 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
hts/BuySellChecker.py
View File

@@ -14,6 +14,8 @@ class BuySellChecker:
macd = None
ichimokuCloud = None
buy_type = -1
def __init__(self):
self.common = Common()
self.stochastic = Stochastic()
@@ -21,6 +23,8 @@ class BuySellChecker:
self.macd = MACD()
self.ichimokuCloud = IchimokuCloud()
self.buy_type = -1
return
@@ -978,8 +982,18 @@ class BuySellChecker:
buy, buy_weight, buy_type = self.getBuyPriceAndWeight(data, last_index)
sell, sell_weight, sell_type = self.getSellPriceAndWeight(data, last_index)
if buy_type == 1:
self.buy_type = 1
if buy_type == 0 or sell_type == 1:
self.buy_type = -1
if self.buy_type == 1:
bsLine['buy'] = [buy]
bsLine['buy_weight'] = [buy_weight]
else:
bsLine['buy'] = [-1]
bsLine['buy_weight'] = [-1]
bsLine['sell'] = [sell]
bsLine['sell_weight'] = [sell_weight]
else:
@@ -989,7 +1003,6 @@ class BuySellChecker:
bsLine['sell'] = [-1 for i in range(size)]
bsLine['sell_weight'] = [-1 for i in range(size)]
self.buy_type = -1
for i in range(size):
buy, buy_weight, buy_type = self.getBuyPriceAndWeight(data, i)
sell, sell_weight, sell_type = self.getSellPriceAndWeight(data, i)