From 9a6e552cb0ccb8fdce09d8374fac21a48a1da147 Mon Sep 17 00:00:00 2001
From: dsyoon <dosangyoon@gmail.com>
Date: Sat, 20 Aug 2022 10:26:15 +0900
Subject: [PATCH] init

---
 stock/util/Stock2Vector.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/stock/util/Stock2Vector.py b/stock/util/Stock2Vector.py
index dc53279..0c0e180 100644
--- a/stock/util/Stock2Vector.py
+++ b/stock/util/Stock2Vector.py
@@ -368,12 +368,12 @@ class Stock2Vector(HTS):
 
         size = len(label)
         batch_X, batch_Y = [], []
-        CHANNEL_SIZE = 4
+        CHANNEL_SIZE = 3
         for i in range(VECTOR_SIZE*CHANNEL_SIZE-1, size):
             X = np.zeros((CHANNEL_SIZE, VECTOR_SIZE, VECTOR_SIZE))
             s = i - VECTOR_SIZE*CHANNEL_SIZE + 1
             e = s+VECTOR_SIZE
-            for c in range(0, 4):
+            for c in range(0, CHANNEL_SIZE):
                 if c > 0:
                     s = e
                     e += VECTOR_SIZE