Change type of branch field in DecisionTree class

Type was Any, now it is DecisionTreeBranch.
This commit is contained in:
Michał Czekański 2020-05-21 16:57:33 +02:00
parent a3d47b4320
commit fd54303dba

View File

@ -1,13 +1,14 @@
from typing import List, Any
from src.AI.DecisionTrees.AttributeDefinition import AttributeDefinition
from src.AI.DecisionTrees.DecisionTreeBranch import DecisionTreeBranch
class DecisionTree(object):
root: AttributeDefinition
branches: List[Any]
branches: List[DecisionTreeBranch]
def __init__(self, root):
self.root = root
self.branches = []
self.branchesNum = 0
self.branchesNum = 0