init

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
-            for c in range(0, 4):
-                if c == 0 :
             s = i - VECTOR_SIZE*CHANNEL_SIZE + 1
-                    e += VECTOR_SIZE
-                else:
+            e = s+VECTOR_SIZE
+            for c in range(0, 4):
+                if c > 0:
                     s = e
-                    e += + VECTOR_SIZE
+                    e += VECTOR_SIZE
 
                 X[c, 0] = macd[s: e]
                 X[c, 1] = diff_avg27[s: e]