Change type of branch field in DecisionTree class
Type was Any, now it is DecisionTreeBranch.
This commit is contained in:
parent
a3d47b4320
commit
fd54303dba
@ -1,11 +1,12 @@
|
||||
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
|
||||
|
Loading…
Reference in New Issue
Block a user