From 7c12e655b45e11854172e56a3c9172a9073f1ab3 Mon Sep 17 00:00:00 2001 From: dsyoon Date: Fri, 19 Aug 2022 22:51:11 +0900 Subject: [PATCH] init --- stock/util/Stock2Vector.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/stock/util/Stock2Vector.py b/stock/util/Stock2Vector.py index 4457bae..dc53279 100644 --- a/stock/util/Stock2Vector.py +++ b/stock/util/Stock2Vector.py @@ -371,14 +371,12 @@ class Stock2Vector(HTS): CHANNEL_SIZE = 4 for i in range(VECTOR_SIZE*CHANNEL_SIZE-1, 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): - if c == 0 : - s = i - VECTOR_SIZE*CHANNEL_SIZE + 1 - e += VECTOR_SIZE - else: + if c > 0: s = e - e += + VECTOR_SIZE + e += VECTOR_SIZE X[c, 0] = macd[s: e] X[c, 1] = diff_avg27[s: e]