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-19 22:51:11 +09:00
parent 3485cafd85
commit 7c12e655b4
1 changed files with 4 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
stock/util/Stock2Vector.py
View File

@@ -371,14 +371,12 @@ class Stock2Vector(HTS):
CHANNEL_SIZE = 4 CHANNEL_SIZE = 4
for i in range(VECTOR_SIZE*CHANNEL_SIZE-1, size): for i in range(VECTOR_SIZE*CHANNEL_SIZE-1, size):
X = np.zeros((CHANNEL_SIZE, VECTOR_SIZE, VECTOR_SIZE)) X = np.zeros((CHANNEL_SIZE, VECTOR_SIZE, VECTOR_SIZE))
s, e = i, i s = i - VECTOR_SIZE*CHANNEL_SIZE + 1
e = s+VECTOR_SIZE
for c in range(0, 4): for c in range(0, 4):
if c == 0 : if c > 0:
s = i - VECTOR_SIZE*CHANNEL_SIZE + 1
e += VECTOR_SIZE
else:
s = e s = e
e += + VECTOR_SIZE e += VECTOR_SIZE
X[c, 0] = macd[s: e] X[c, 0] = macd[s: e]
X[c, 1] = diff_avg27[s: e] X[c, 1] = diff_avg27[s: e]