From f7cccd4b4d62e14d8bc6c1bc4070694fd1081835 Mon Sep 17 00:00:00 2001 From: xVulpeSx Date: Thu, 12 May 2022 19:52:13 +0200 Subject: [PATCH] decisionTree final update --- data/TEST/importedData.csv | 289 ++++++++++++++++++++++++++++++------- main.py | 17 ++- 2 files changed, 251 insertions(+), 55 deletions(-) diff --git a/data/TEST/importedData.csv b/data/TEST/importedData.csv index e13bd0d..c2c1827 100644 --- a/data/TEST/importedData.csv +++ b/data/TEST/importedData.csv @@ -1,51 +1,240 @@ DELAY,PAYED,NET-WORTH,INFLUENCE,SKARBOWKA,MEMBER,HAT,SIZE,PRIORITY -11,FALSE,41,97,TRUE,FALSE,TRUE,CompanySize.HUGE,LOW -7,FALSE,22,80,TRUE,FALSE,FALSE,CompanySize.NORMAL,LOW -3,FALSE,58,0,TRUE,TRUE,TRUE,CompanySize.BIG,LOW -11,FALSE,3,15,FALSE,TRUE,FALSE,CompanySize.NO,LOW -5,FALSE,42,18,TRUE,FALSE,FALSE,CompanySize.SMALL,LOW -4,TRUE,51,54,TRUE,FALSE,TRUE,CompanySize.NORMAL,HIGH -7,TRUE,18,47,FALSE,TRUE,TRUE,CompanySize.SMALL,LOW -3,TRUE,96,61,FALSE,TRUE,TRUE,CompanySize.SMALL,MEDIUM -13,FALSE,42,44,TRUE,TRUE,TRUE,CompanySize.NORMAL,LOW -4,TRUE,6,3,FALSE,FALSE,FALSE,CompanySize.NORMAL,LOW -10,TRUE,91,36,FALSE,FALSE,TRUE,CompanySize.NO,MEDIUM -1,TRUE,80,11,TRUE,FALSE,TRUE,CompanySize.GIGANTISHE,HIGH -6,FALSE,91,82,FALSE,TRUE,TRUE,CompanySize.NO,LOW -4,FALSE,1,93,TRUE,TRUE,FALSE,CompanySize.BIG,LOW -14,FALSE,67,13,TRUE,TRUE,TRUE,CompanySize.SMALL,LOW -0,FALSE,7,58,FALSE,FALSE,FALSE,CompanySize.NORMAL,LOW -8,TRUE,74,67,TRUE,TRUE,TRUE,CompanySize.NORMAL,HIGH -4,TRUE,33,43,FALSE,TRUE,FALSE,CompanySize.BIG,MEDIUM -8,TRUE,74,44,TRUE,FALSE,TRUE,CompanySize.HUGE,HIGH -14,FALSE,59,33,TRUE,FALSE,FALSE,CompanySize.NORMAL,LOW -6,FALSE,87,80,TRUE,TRUE,FALSE,CompanySize.GIGANTISHE,LOW -10,FALSE,2,45,FALSE,FALSE,FALSE,CompanySize.BIG,LOW -7,FALSE,74,17,TRUE,FALSE,FALSE,CompanySize.SMALL,LOW -14,FALSE,14,80,FALSE,TRUE,FALSE,CompanySize.NO,LOW -1,FALSE,74,82,TRUE,TRUE,FALSE,CompanySize.NO,LOW -13,FALSE,66,50,FALSE,TRUE,TRUE,CompanySize.HUGE,LOW -12,TRUE,55,82,TRUE,TRUE,TRUE,CompanySize.NO,HIGH -0,TRUE,63,1,TRUE,TRUE,TRUE,CompanySize.NO,HIGH -0,FALSE,39,70,FALSE,FALSE,TRUE,CompanySize.NORMAL,LOW -1,FALSE,14,66,FALSE,FALSE,FALSE,CompanySize.BIG,LOW -7,FALSE,48,86,TRUE,TRUE,TRUE,CompanySize.BIG,LOW -7,FALSE,39,41,FALSE,TRUE,FALSE,CompanySize.HUGE,LOW -6,TRUE,29,90,FALSE,FALSE,FALSE,CompanySize.GIGANTISHE,HIGH -8,TRUE,79,49,FALSE,TRUE,FALSE,CompanySize.BIG,HIGH -14,TRUE,51,51,FALSE,TRUE,FALSE,CompanySize.NO,LOW -1,FALSE,92,97,FALSE,TRUE,TRUE,CompanySize.HUGE,LOW -6,FALSE,92,90,TRUE,FALSE,FALSE,CompanySize.GIGANTISHE,LOW -9,FALSE,89,34,FALSE,TRUE,TRUE,CompanySize.NO,LOW -14,FALSE,85,8,FALSE,FALSE,TRUE,CompanySize.HUGE,LOW -14,TRUE,86,30,FALSE,TRUE,FALSE,CompanySize.GIGANTISHE,MEDIUM -3,TRUE,82,57,FALSE,TRUE,FALSE,CompanySize.BIG,MEDIUM -8,TRUE,18,44,FALSE,TRUE,FALSE,CompanySize.HUGE,LOW -0,FALSE,87,32,FALSE,FALSE,FALSE,CompanySize.NO,LOW -10,TRUE,97,26,FALSE,TRUE,TRUE,CompanySize.HUGE,HIGH -0,FALSE,88,98,FALSE,TRUE,FALSE,CompanySize.NO,LOW -10,TRUE,27,82,FALSE,FALSE,FALSE,CompanySize.HUGE,MEDIUM -8,TRUE,28,36,TRUE,FALSE,FALSE,CompanySize.HUGE,HIGH -14,FALSE,48,94,TRUE,FALSE,TRUE,CompanySize.HUGE,LOW -7,FALSE,40,63,TRUE,TRUE,TRUE,CompanySize.BIG,LOW -8,FALSE,90,20,TRUE,TRUE,FALSE,CompanySize.NO,LOW \ No newline at end of file +13,TRUE,13,20,FALSE,FALSE,TRUE,CompanySize.BIG,MEDIUM +3,FALSE,93,31,FALSE,TRUE,FALSE,CompanySize.SMALL,LOW +2,TRUE,8,93,FALSE,FALSE,FALSE,CompanySize.SMALL,MEDIUM +8,TRUE,43,77,TRUE,TRUE,TRUE,CompanySize.SMALL,HIGH +5,FALSE,79,71,FALSE,TRUE,TRUE,CompanySize.BIG,LOW +2,TRUE,14,46,FALSE,FALSE,FALSE,CompanySize.NORMAL,LOW +0,FALSE,18,16,TRUE,FALSE,TRUE,CompanySize.GIGANTISHE,HIGH +9,TRUE,33,79,FALSE,FALSE,TRUE,CompanySize.HUGE,HIGH +2,TRUE,38,48,FALSE,TRUE,TRUE,CompanySize.HUGE,HIGH +0,TRUE,79,66,FALSE,FALSE,FALSE,CompanySize.SMALL,MEDIUM +6,FALSE,21,33,FALSE,FALSE,FALSE,CompanySize.NORMAL,LOW +1,TRUE,79,43,FALSE,TRUE,TRUE,CompanySize.HUGE,HIGH +0,TRUE,42,60,TRUE,TRUE,TRUE,CompanySize.SMALL,HIGH +1,FALSE,59,25,TRUE,TRUE,TRUE,CompanySize.GIGANTISHE,HIGH +3,TRUE,88,50,FALSE,FALSE,TRUE,CompanySize.HUGE,HIGH +6,FALSE,0,4,FALSE,TRUE,TRUE,CompanySize.NO,LOW +13,FALSE,67,9,FALSE,TRUE,FALSE,CompanySize.NORMAL,LOW +10,TRUE,53,38,FALSE,TRUE,FALSE,CompanySize.NORMAL,MEDIUM +5,TRUE,26,64,FALSE,TRUE,TRUE,CompanySize.SMALL,MEDIUM +9,TRUE,32,79,FALSE,TRUE,FALSE,CompanySize.GIGANTISHE,MEDIUM +13,TRUE,95,94,FALSE,TRUE,FALSE,CompanySize.GIGANTISHE,HIGH +10,TRUE,0,58,FALSE,FALSE,FALSE,CompanySize.GIGANTISHE,MEDIUM +6,TRUE,86,91,FALSE,TRUE,TRUE,CompanySize.BIG,HIGH +8,TRUE,52,5,FALSE,FALSE,FALSE,CompanySize.GIGANTISHE,MEDIUM +8,TRUE,1,46,FALSE,FALSE,FALSE,CompanySize.HUGE,MEDIUM +6,TRUE,80,10,FALSE,TRUE,TRUE,CompanySize.NORMAL,HIGH +4,FALSE,29,53,FALSE,TRUE,TRUE,CompanySize.BIG,LOW +3,TRUE,49,35,TRUE,FALSE,TRUE,CompanySize.NORMAL,HIGH +4,TRUE,38,1,FALSE,TRUE,FALSE,CompanySize.BIG,MEDIUM +8,FALSE,88,96,FALSE,FALSE,FALSE,CompanySize.NO,LOW +6,TRUE,64,27,FALSE,FALSE,FALSE,CompanySize.HUGE,MEDIUM +5,TRUE,87,50,FALSE,FALSE,FALSE,CompanySize.BIG,MEDIUM +1,FALSE,87,29,TRUE,TRUE,TRUE,CompanySize.SMALL,HIGH +3,FALSE,63,52,FALSE,FALSE,FALSE,CompanySize.BIG,LOW +14,FALSE,74,33,FALSE,FALSE,FALSE,CompanySize.HUGE,LOW +10,TRUE,81,22,FALSE,FALSE,FALSE,CompanySize.NORMAL,LOW +14,TRUE,78,85,TRUE,TRUE,FALSE,CompanySize.HUGE,HIGH +0,TRUE,76,91,FALSE,TRUE,TRUE,CompanySize.GIGANTISHE,HIGH +8,TRUE,54,11,FALSE,FALSE,TRUE,CompanySize.BIG,MEDIUM +8,TRUE,52,92,FALSE,TRUE,FALSE,CompanySize.GIGANTISHE,MEDIUM +0,TRUE,86,29,FALSE,TRUE,FALSE,CompanySize.SMALL,LOW +9,TRUE,15,94,TRUE,FALSE,TRUE,CompanySize.HUGE,HIGH +14,FALSE,90,84,FALSE,TRUE,TRUE,CompanySize.NO,LOW +3,TRUE,68,30,FALSE,TRUE,TRUE,CompanySize.GIGANTISHE,HIGH +9,TRUE,14,39,FALSE,TRUE,TRUE,CompanySize.BIG,MEDIUM +5,TRUE,19,40,FALSE,FALSE,FALSE,CompanySize.NO,LOW +1,TRUE,17,71,FALSE,TRUE,FALSE,CompanySize.HUGE,MEDIUM +0,FALSE,41,97,FALSE,FALSE,TRUE,CompanySize.HUGE,LOW +2,FALSE,22,3,FALSE,TRUE,FALSE,CompanySize.NO,LOW +8,TRUE,14,64,FALSE,FALSE,TRUE,CompanySize.NORMAL,MEDIUM +4,TRUE,10,46,FALSE,FALSE,TRUE,CompanySize.NO,LOW +6,TRUE,39,75,FALSE,FALSE,FALSE,CompanySize.BIG,LOW +6,TRUE,96,68,FALSE,FALSE,TRUE,CompanySize.HUGE,HIGH +5,TRUE,60,10,TRUE,FALSE,TRUE,CompanySize.HUGE,HIGH +13,TRUE,6,73,FALSE,TRUE,FALSE,CompanySize.SMALL,LOW +2,FALSE,21,10,FALSE,TRUE,TRUE,CompanySize.GIGANTISHE,LOW +4,FALSE,34,59,FALSE,FALSE,TRUE,CompanySize.SMALL,LOW +11,TRUE,5,57,TRUE,FALSE,FALSE,CompanySize.HUGE,HIGH +5,TRUE,84,34,FALSE,FALSE,TRUE,CompanySize.GIGANTISHE,MEDIUM +7,TRUE,23,23,FALSE,TRUE,TRUE,CompanySize.HUGE,MEDIUM +1,TRUE,10,38,TRUE,TRUE,FALSE,CompanySize.SMALL,HIGH +3,TRUE,36,89,TRUE,TRUE,TRUE,CompanySize.BIG,HIGH +1,TRUE,12,33,FALSE,TRUE,TRUE,CompanySize.BIG,MEDIUM +4,FALSE,21,72,FALSE,FALSE,FALSE,CompanySize.NO,LOW +2,FALSE,34,44,FALSE,FALSE,FALSE,CompanySize.GIGANTISHE,LOW +8,TRUE,25,29,FALSE,TRUE,TRUE,CompanySize.HUGE,MEDIUM +2,TRUE,96,94,FALSE,FALSE,FALSE,CompanySize.NORMAL,MEDIUM +13,TRUE,91,0,FALSE,FALSE,TRUE,CompanySize.NO,LOW +13,TRUE,26,24,FALSE,FALSE,TRUE,CompanySize.BIG,MEDIUM +3,FALSE,96,43,TRUE,TRUE,FALSE,CompanySize.BIG,HIGH +12,TRUE,48,97,FALSE,FALSE,TRUE,CompanySize.SMALL,HIGH +10,TRUE,23,5,FALSE,TRUE,FALSE,CompanySize.NO,LOW +4,FALSE,77,98,FALSE,FALSE,TRUE,CompanySize.GIGANTISHE,LOW +4,FALSE,95,39,FALSE,FALSE,FALSE,CompanySize.HUGE,LOW +1,TRUE,33,40,FALSE,TRUE,TRUE,CompanySize.GIGANTISHE,MEDIUM +9,TRUE,2,11,FALSE,FALSE,FALSE,CompanySize.BIG,LOW +6,TRUE,90,62,TRUE,TRUE,FALSE,CompanySize.HUGE,HIGH +13,TRUE,53,62,FALSE,FALSE,FALSE,CompanySize.NORMAL,MEDIUM +9,TRUE,3,15,FALSE,TRUE,FALSE,CompanySize.NORMAL,LOW +7,TRUE,5,2,FALSE,TRUE,FALSE,CompanySize.NORMAL,LOW +6,TRUE,14,59,FALSE,FALSE,FALSE,CompanySize.BIG,MEDIUM +10,TRUE,93,58,FALSE,FALSE,FALSE,CompanySize.NORMAL,MEDIUM +8,TRUE,95,23,FALSE,TRUE,TRUE,CompanySize.SMALL,MEDIUM +14,TRUE,6,37,FALSE,TRUE,TRUE,CompanySize.HUGE,MEDIUM +9,TRUE,71,3,FALSE,TRUE,TRUE,CompanySize.GIGANTISHE,HIGH +8,TRUE,4,5,FALSE,FALSE,FALSE,CompanySize.SMALL,LOW +0,FALSE,32,8,FALSE,TRUE,FALSE,CompanySize.NORMAL,LOW +8,TRUE,20,43,FALSE,TRUE,TRUE,CompanySize.HUGE,MEDIUM +13,TRUE,50,44,FALSE,TRUE,TRUE,CompanySize.HUGE,MEDIUM +11,TRUE,59,93,FALSE,FALSE,FALSE,CompanySize.NO,MEDIUM +4,TRUE,65,34,FALSE,TRUE,TRUE,CompanySize.GIGANTISHE,LOW +12,FALSE,47,100,FALSE,TRUE,TRUE,CompanySize.GIGANTISHE,LOW +10,TRUE,65,97,FALSE,FALSE,TRUE,CompanySize.NORMAL,MEDIUM +9,FALSE,81,42,FALSE,FALSE,FALSE,CompanySize.GIGANTISHE,LOW +13,FALSE,28,21,FALSE,FALSE,FALSE,CompanySize.NO,LOW +9,FALSE,93,93,FALSE,FALSE,FALSE,CompanySize.BIG,LOW +7,TRUE,64,26,TRUE,TRUE,TRUE,CompanySize.HUGE,HIGH +10,FALSE,28,72,FALSE,TRUE,FALSE,CompanySize.HUGE,LOW +0,FALSE,34,50,FALSE,TRUE,TRUE,CompanySize.HUGE,LOW +9,FALSE,36,44,FALSE,FALSE,FALSE,CompanySize.HUGE,LOW +11,TRUE,63,1,FALSE,TRUE,TRUE,CompanySize.HUGE,MEDIUM +8,FALSE,38,34,FALSE,TRUE,FALSE,CompanySize.SMALL,LOW +3,TRUE,57,7,FALSE,FALSE,TRUE,CompanySize.HUGE,MEDIUM +14,FALSE,12,23,TRUE,FALSE,TRUE,CompanySize.NO,HIGH +4,TRUE,16,98,FALSE,TRUE,FALSE,CompanySize.GIGANTISHE,HIGH +5,TRUE,39,69,FALSE,FALSE,FALSE,CompanySize.NO,LOW +0,TRUE,82,77,TRUE,FALSE,FALSE,CompanySize.GIGANTISHE,HIGH +11,TRUE,84,64,FALSE,TRUE,TRUE,CompanySize.GIGANTISHE,HIGH +12,TRUE,71,77,TRUE,TRUE,FALSE,CompanySize.BIG,HIGH +8,TRUE,42,54,FALSE,TRUE,TRUE,CompanySize.GIGANTISHE,HIGH +1,TRUE,23,93,FALSE,FALSE,FALSE,CompanySize.NO,MEDIUM +1,TRUE,38,40,FALSE,TRUE,TRUE,CompanySize.NORMAL,MEDIUM +1,FALSE,33,11,FALSE,TRUE,TRUE,CompanySize.BIG,LOW +0,TRUE,5,56,FALSE,TRUE,TRUE,CompanySize.NO,MEDIUM +13,TRUE,49,71,FALSE,FALSE,TRUE,CompanySize.NO,MEDIUM +9,TRUE,64,29,FALSE,FALSE,TRUE,CompanySize.BIG,MEDIUM +8,TRUE,44,95,FALSE,TRUE,FALSE,CompanySize.BIG,MEDIUM +0,TRUE,40,3,FALSE,FALSE,FALSE,CompanySize.SMALL,LOW +2,TRUE,49,39,FALSE,TRUE,TRUE,CompanySize.NO,LOW +10,TRUE,4,94,FALSE,TRUE,TRUE,CompanySize.BIG,HIGH +0,TRUE,90,86,FALSE,TRUE,FALSE,CompanySize.SMALL,MEDIUM +14,TRUE,61,74,FALSE,FALSE,FALSE,CompanySize.GIGANTISHE,MEDIUM +3,TRUE,86,27,FALSE,FALSE,TRUE,CompanySize.NORMAL,MEDIUM +0,FALSE,47,19,FALSE,TRUE,FALSE,CompanySize.SMALL,LOW +7,FALSE,60,30,FALSE,TRUE,FALSE,CompanySize.NORMAL,LOW +13,TRUE,96,89,FALSE,FALSE,FALSE,CompanySize.NO,MEDIUM +7,TRUE,83,73,FALSE,TRUE,TRUE,CompanySize.HUGE,HIGH +7,TRUE,21,30,FALSE,FALSE,TRUE,CompanySize.NO,LOW +10,TRUE,26,41,FALSE,TRUE,TRUE,CompanySize.NO,MEDIUM +13,FALSE,73,16,FALSE,TRUE,FALSE,CompanySize.NO,LOW +12,TRUE,30,54,TRUE,TRUE,TRUE,CompanySize.GIGANTISHE,HIGH +12,FALSE,9,52,FALSE,FALSE,FALSE,CompanySize.NORMAL,LOW +12,TRUE,28,10,FALSE,FALSE,FALSE,CompanySize.SMALL,LOW +2,FALSE,23,34,FALSE,TRUE,FALSE,CompanySize.NORMAL,LOW +6,TRUE,50,85,FALSE,TRUE,FALSE,CompanySize.NO,MEDIUM +10,TRUE,86,74,FALSE,TRUE,TRUE,CompanySize.SMALL,MEDIUM +10,TRUE,4,39,FALSE,FALSE,TRUE,CompanySize.GIGANTISHE,HIGH +3,TRUE,11,66,FALSE,TRUE,FALSE,CompanySize.NORMAL,MEDIUM +6,FALSE,31,1,FALSE,FALSE,TRUE,CompanySize.NO,LOW +8,TRUE,77,29,FALSE,FALSE,TRUE,CompanySize.BIG,MEDIUM +0,TRUE,85,91,FALSE,FALSE,FALSE,CompanySize.NO,MEDIUM +0,FALSE,51,68,FALSE,FALSE,FALSE,CompanySize.NORMAL,LOW +6,TRUE,54,32,FALSE,FALSE,TRUE,CompanySize.BIG,MEDIUM +8,FALSE,2,78,FALSE,FALSE,FALSE,CompanySize.HUGE,LOW +0,FALSE,92,47,FALSE,TRUE,FALSE,CompanySize.NORMAL,LOW +0,TRUE,9,34,TRUE,FALSE,TRUE,CompanySize.GIGANTISHE,HIGH +11,TRUE,3,40,FALSE,TRUE,TRUE,CompanySize.HUGE,HIGH +5,TRUE,6,58,FALSE,TRUE,TRUE,CompanySize.HUGE,HIGH +5,TRUE,27,61,FALSE,FALSE,TRUE,CompanySize.NO,LOW +2,TRUE,70,98,FALSE,TRUE,FALSE,CompanySize.NO,MEDIUM +6,TRUE,2,11,FALSE,FALSE,FALSE,CompanySize.HUGE,MEDIUM +3,TRUE,64,14,FALSE,TRUE,FALSE,CompanySize.NORMAL,MEDIUM +5,TRUE,63,84,FALSE,TRUE,TRUE,CompanySize.SMALL,HIGH +5,TRUE,72,32,FALSE,FALSE,TRUE,CompanySize.NORMAL,MEDIUM +8,TRUE,48,7,TRUE,TRUE,FALSE,CompanySize.GIGANTISHE,HIGH +11,FALSE,4,96,TRUE,FALSE,FALSE,CompanySize.NORMAL,HIGH +8,TRUE,60,76,FALSE,FALSE,FALSE,CompanySize.SMALL,LOW +4,TRUE,17,24,FALSE,FALSE,TRUE,CompanySize.GIGANTISHE,MEDIUM +9,TRUE,59,88,FALSE,TRUE,FALSE,CompanySize.NORMAL,HIGH +10,TRUE,95,79,FALSE,TRUE,FALSE,CompanySize.SMALL,MEDIUM +2,TRUE,59,20,FALSE,TRUE,FALSE,CompanySize.NO,LOW +7,TRUE,75,61,FALSE,FALSE,FALSE,CompanySize.GIGANTISHE,HIGH +1,TRUE,10,89,TRUE,FALSE,TRUE,CompanySize.GIGANTISHE,HIGH +6,TRUE,69,78,TRUE,TRUE,TRUE,CompanySize.NO,HIGH +5,TRUE,98,43,FALSE,TRUE,TRUE,CompanySize.GIGANTISHE,HIGH +4,TRUE,14,1,FALSE,TRUE,FALSE,CompanySize.SMALL,LOW +3,FALSE,56,85,FALSE,TRUE,TRUE,CompanySize.GIGANTISHE,LOW +0,TRUE,22,0,FALSE,FALSE,FALSE,CompanySize.BIG,MEDIUM +2,TRUE,28,61,FALSE,FALSE,FALSE,CompanySize.BIG,HIGH +14,TRUE,99,34,FALSE,FALSE,FALSE,CompanySize.GIGANTISHE,HIGH +3,TRUE,90,96,FALSE,TRUE,TRUE,CompanySize.BIG,HIGH +1,TRUE,25,98,FALSE,FALSE,TRUE,CompanySize.HUGE,HIGH +5,TRUE,16,49,FALSE,TRUE,TRUE,CompanySize.NORMAL,MEDIUM +0,TRUE,40,97,FALSE,TRUE,FALSE,CompanySize.SMALL,MEDIUM +0,TRUE,31,17,FALSE,TRUE,TRUE,CompanySize.NORMAL,MEDIUM +1,TRUE,43,4,FALSE,FALSE,TRUE,CompanySize.NORMAL,MEDIUM +5,TRUE,69,65,FALSE,TRUE,FALSE,CompanySize.SMALL,LOW +2,TRUE,45,9,FALSE,FALSE,TRUE,CompanySize.NORMAL,MEDIUM +5,FALSE,24,41,FALSE,FALSE,FALSE,CompanySize.BIG,LOW +0,TRUE,0,25,FALSE,TRUE,FALSE,CompanySize.BIG,MEDIUM +13,TRUE,41,91,TRUE,FALSE,TRUE,CompanySize.NORMAL,HIGH +5,TRUE,15,10,FALSE,TRUE,TRUE,CompanySize.GIGANTISHE,HIGH +5,TRUE,91,94,FALSE,TRUE,TRUE,CompanySize.HUGE,HIGH +9,TRUE,83,98,FALSE,TRUE,FALSE,CompanySize.HUGE,HIGH +12,TRUE,58,56,FALSE,FALSE,FALSE,CompanySize.BIG,MEDIUM +8,TRUE,78,39,FALSE,TRUE,FALSE,CompanySize.BIG,HIGH +12,TRUE,8,80,FALSE,TRUE,FALSE,CompanySize.NORMAL,MEDIUM +3,TRUE,80,52,FALSE,FALSE,TRUE,CompanySize.NORMAL,MEDIUM +12,TRUE,23,47,FALSE,TRUE,FALSE,CompanySize.BIG,MEDIUM +6,FALSE,21,94,TRUE,TRUE,FALSE,CompanySize.GIGANTISHE,HIGH +3,TRUE,32,58,FALSE,FALSE,FALSE,CompanySize.HUGE,MEDIUM +3,TRUE,74,26,FALSE,TRUE,FALSE,CompanySize.BIG,MEDIUM +0,TRUE,7,2,FALSE,TRUE,TRUE,CompanySize.BIG,MEDIUM +7,TRUE,90,32,FALSE,FALSE,FALSE,CompanySize.NO,LOW +8,TRUE,81,21,TRUE,FALSE,TRUE,CompanySize.NORMAL,HIGH +14,TRUE,84,54,FALSE,FALSE,FALSE,CompanySize.GIGANTISHE,HIGH +5,TRUE,99,96,FALSE,FALSE,TRUE,CompanySize.NO,MEDIUM +14,TRUE,93,12,TRUE,TRUE,FALSE,CompanySize.HUGE,HIGH +5,TRUE,17,34,FALSE,FALSE,FALSE,CompanySize.SMALL,LOW +3,TRUE,78,76,FALSE,TRUE,TRUE,CompanySize.GIGANTISHE,HIGH +9,FALSE,37,65,FALSE,TRUE,FALSE,CompanySize.GIGANTISHE,LOW +12,FALSE,61,84,FALSE,FALSE,FALSE,CompanySize.NO,LOW +11,FALSE,72,6,FALSE,FALSE,TRUE,CompanySize.GIGANTISHE,LOW +1,TRUE,54,38,FALSE,FALSE,TRUE,CompanySize.GIGANTISHE,MEDIUM +9,FALSE,35,72,FALSE,TRUE,FALSE,CompanySize.NORMAL,LOW +6,TRUE,77,86,FALSE,FALSE,FALSE,CompanySize.NO,MEDIUM +7,TRUE,52,75,FALSE,FALSE,FALSE,CompanySize.NORMAL,MEDIUM +5,TRUE,50,14,FALSE,FALSE,TRUE,CompanySize.GIGANTISHE,HIGH +13,FALSE,88,14,FALSE,FALSE,TRUE,CompanySize.BIG,LOW +13,FALSE,81,15,FALSE,FALSE,TRUE,CompanySize.GIGANTISHE,LOW +9,FALSE,99,73,FALSE,FALSE,FALSE,CompanySize.BIG,LOW +9,FALSE,18,50,FALSE,TRUE,FALSE,CompanySize.NO,LOW +9,FALSE,51,2,FALSE,FALSE,TRUE,CompanySize.NO,LOW +6,TRUE,30,68,FALSE,TRUE,FALSE,CompanySize.BIG,MEDIUM +7,FALSE,24,12,FALSE,TRUE,TRUE,CompanySize.GIGANTISHE,LOW +7,FALSE,22,3,FALSE,TRUE,TRUE,CompanySize.NORMAL,LOW +5,TRUE,95,61,FALSE,FALSE,FALSE,CompanySize.HUGE,LOW +12,FALSE,68,33,FALSE,FALSE,TRUE,CompanySize.NORMAL,LOW +7,TRUE,12,25,FALSE,TRUE,TRUE,CompanySize.NORMAL,LOW +8,FALSE,63,4,FALSE,FALSE,TRUE,CompanySize.GIGANTISHE,LOW +8,TRUE,82,18,FALSE,FALSE,TRUE,CompanySize.HUGE,LOW +5,TRUE,56,85,FALSE,FALSE,FALSE,CompanySize.HUGE,LOW +7,FALSE,96,29,FALSE,FALSE,TRUE,CompanySize.GIGANTISHE,LOW +9,FALSE,24,46,FALSE,TRUE,TRUE,CompanySize.BIG,LOW +9,FALSE,38,35,FALSE,TRUE,TRUE,CompanySize.NO,LOW +5,TRUE,52,89,FALSE,FALSE,FALSE,CompanySize.HUGE,MEDIUM +12,FALSE,94,52,FALSE,FALSE,FALSE,CompanySize.BIG,LOW +7,TRUE,36,27,FALSE,TRUE,TRUE,CompanySize.GIGANTISHE,MEDIUM +7,FALSE,78,76,FALSE,FALSE,FALSE,CompanySize.HUGE,LOW +8,FALSE,83,93,TRUE,FALSE,FALSE,CompanySize.NO,HIGH +9,FALSE,47,66,FALSE,TRUE,FALSE,CompanySize.HUGE,LOW +11,FALSE,39,78,FALSE,TRUE,FALSE,CompanySize.GIGANTISHE,LOW +9,FALSE,8,41,FALSE,TRUE,TRUE,CompanySize.NORMAL,LOW +10,FALSE,86,97,TRUE,FALSE,FALSE,CompanySize.HUGE,HIGH +7,TRUE,85,98,TRUE,FALSE,FALSE,CompanySize.GIGANTISHE,HIGH +11,FALSE,9,25,FALSE,TRUE,TRUE,CompanySize.NORMAL,LOW +3,TRUE,11,57,FALSE,TRUE,FALSE,CompanySize.BIG,LOW +13,FALSE,72,4,FALSE,FALSE,TRUE,CompanySize.NO,LOW +8,TRUE,24,28,FALSE,TRUE,TRUE,CompanySize.NO,LOW \ No newline at end of file diff --git a/main.py b/main.py index 21b4e31..5cd0fa9 100644 --- a/main.py +++ b/main.py @@ -2,6 +2,7 @@ import csv import random import pandas +import sklearn.tree from mesa.visualization.ModularVisualization import ModularServer from mesa.visualization.modules import CanvasGrid @@ -91,25 +92,31 @@ if __name__ == '__main__': data_input = pandas.read_csv('data/TEST/importedData.csv', delimiter=",") - X = data_input[['DELAY','PAYED','NET-WORTH','INFLUENCE','SKARBOWKA','MEMBER','HAT','SIZE']].values + X_headers = ['DELAY','PAYED','NET-WORTH','INFLUENCE','SKARBOWKA','MEMBER','HAT','SIZE'] + + X = data_input[X_headers].values Y = data_input["PRIORITY"] label_BP = preprocessing.LabelEncoder() label_BP.fit(['CompanySize.NO', 'CompanySize.SMALL', 'CompanySize.NORMAL', 'CompanySize.BIG', 'CompanySize.HUGE', 'CompanySize.GIGANTISHE']) X[:, 7] = label_BP.transform(X[:, 7]) - X_train, X_test, y_train, y_test = train_test_split(X, Y, test_size=0.1, train_size=0.9) + X_train, X_test, y_train, y_test = train_test_split(X, Y, test_size=0.2, train_size=0.8) drugTree = DecisionTreeClassifier(criterion="entropy", max_depth=4) - drugTree.fit(X_train, y_train) + clf = drugTree.fit(X_train, y_train) predicted = drugTree.predict(X_test) - print(X_test) - print(predicted) + y_test = y_test.to_list() + + for i in range(len(predicted)): + print("{}. {} \n predicted: {}, actual: {}".format(i, X_test[i,:], predicted[i], y_test[i])) print("\nDecisionTrees's Accuracy: ", metrics.accuracy_score(y_test, predicted)) + print(sklearn.tree.export_text(clf, feature_names=X_headers)) + # base = 512 # gridWidth = 10 # gridHeight = 10