def dtreePathGeneration(self):
dtreePaths = Paths()
newPath = Path()
newPath.addNode(self.dtree.getNode(1))
newPath.addNode(self.dtree.getNode(2))
newPath.addNode(self.dtree.getNode(3))
dtreePaths.addPath(newPath)
newPath = Path()
newPath.addNode(self.dtree.getNode(1))
newPath.addNode(self.dtree.getNode(2))
newPath.addNode(self.dtree.getNode(8))
dtreePaths.addPath(newPath)
newPath = Path()
newPath.addNode(self.dtree.getNode(1))
newPath.addNode(self.dtree.getNode(2))
newPath.addNode(self.dtree.getNode(4))
newPath.addNode(self.dtree.getNode(5))
dtreePaths.addPath(newPath)
newPath = Path()
newPath.addNode(self.dtree.getNode(1))
newPath.addNode(self.dtree.getNode(2))
newPath.addNode(self.dtree.getNode(3))
newPath.addNode(self.dtree.getNode(4))
newPath.addNode(self.dtree.getNode(6))
newPath.addNode(self.dtree.getNode(7))
dtreePaths.addPath(newPath)
return dtreePaths
def _renderPaths(self, paths):
print("paths:", paths)
path = Path("P1", paths[0], self.model.robots[0].color)
self.model.entities[path.name] = path
path.createShape(self)
path = Path("P2", paths[1], self.model.robots[1].color)
self.model.entities[path.name] = path
path.createShape(self)
def __init__(self):
self.path = Path()
self.event_earning_rate_dict = {
'kodex_200': 0, # KODEX 200
'kodex_us_S_and_P500': 0, # KODEX 미국S&P500선물(H)
'tiger_euro_50': 0, # TIGER 유로스탁스50(합성 H)
'kodex_japan_topix_100': 0, # KODEX 일본TOPIX100
}
# NOTE: 예금 금리는 1.5 % 로 고정, 추후 필요하면 Crawler 에서 금리 긁어오도록 수정
self.deposit_interest_rate = 1.5
def decisionTreePathGeneration(self):
decisionTreePaths = Paths()
newPath = Path()
newPath.addNode(self.decisionTree.getNode(1))
newPath.addNode(self.decisionTree.getNode(2))
newPath.addNode(self.decisionTree.getNode(3))
decisionTreePaths.addPath(newPath)
newPath = Path()
newPath.addNode(self.decisionTree.getNode(1))
newPath.addNode(self.decisionTree.getNode(2))
newPath.addNode(self.decisionTree.getNode(6))
decisionTreePaths.addPath(newPath)
newPath = Path()
newPath.addNode(self.decisionTree.getNode(1))
newPath.addNode(self.decisionTree.getNode(2))
newPath.addNode(self.decisionTree.getNode(4))
newPath.addNode(self.decisionTree.getNode(5))
decisionTreePaths.addPath(newPath)
return decisionTreePaths
def generate_new_path(self, startNode):
newPath = Path()
self.generate_path_for_node(startNode, newPath)
return newPath
def add_paths(self, path_list):
paths = [Path(node_a, node_b) for node_a, node_b in path_list]
self._paths.extend(paths)
return paths
