This commit is contained in:
barmal4 2021-05-18 19:36:41 +02:00
commit 520f398e36
6 changed files with 4 additions and 0 deletions

View File

@ -8,12 +8,14 @@ LEAVES = 'action'
class DecisionTree:
def __init__(self, doCreation):
if doCreation:
return
with open('DecisionTree.json', 'r') as fp:
self.tree = json.load(fp)
def find_entropy(self,df):
Class = df.keys()[-1] # To make the code generic, changing target variable class name
entropy = 0
@ -120,3 +122,5 @@ if __name__ == "__main__":

BIN
Engine/go_to_poligon.png Normal file

Binary file not shown.

After

Width:  |  Height:  |  Size: 20 KiB