init

stock/util/LabelMaker.py

@@ -30,7 +30,7 @@ class LabelMaker (HTS):
 
         # 매수 처리
         p_min_price = 999999999
-        # - 10을 한 것은 3시 11분까지 매매를 하기 위함임
+        # - 10을 한 것은 3시 11분까지 매수를 하기 위함임
         for i in range(size-duration-10):
             min_price, min_price_c, = 9999999, -1
             for c in range(i, i+duration):
@@ -71,7 +71,8 @@ class LabelMaker (HTS):
                                 bsLine['sell'][c] = max_price
                                 bsLine['sell_weight'][c] = 1
             else:
-                boundry = data["high"][buy_i[i] + 1:]
+                # - 5를 한 것은 3시 16분까지 매도를 하기 위함임
+                boundry = data["high"][buy_i[i] + 1: len(data["high"])-5]
                 if len(boundry) > 1:
                     max_price = max(boundry)
                     for c in range(buy_i[i], len(data["high"])):
@@ -270,7 +271,7 @@ if __name__ == "__main__":
 
     labelMaker = LabelMaker()
 
-    view = True
+    view = False
     stock_code = "252670"
     ymd = "20200701"