s444409/DSZI_Survival
s444409/DSZI_Survival
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Change type of branch field in DecisionTree class

Browse Source 
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
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/AI/DecisionTrees/DecisionTree.py
View File

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