s434713/SZI_PROJEKT_GR1_TRAKTOR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

poprawny porogram

Browse Source
This commit is contained in:
Paulina J 2019-06-10 11:26:22 +02:00
parent 93ac9d9df5
commit f9f774ba51
3 changed files with 11 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
Tree1.py
View File

@ -31,13 +31,11 @@ class TreeClass:
with open('dane.txt') as data: with open('dane.txt') as data:
for line in data: for line in data:
elements = line.split(", ") elements = line.split(", ")
for element in elements: if elements[1] == "True":
if element == "True": elements[1] = 1
element = True if elements[1] == "False":
elif element == "False": elements[1] = 0
element = False
else:
element = int(element)
features.append(elements) features.append(elements)
labels = [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, labels = [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,

2
dane.txt
View File

@ -273,5 +273,3 @@
0, False, 230, 850, 180 0, False, 230, 850, 180
0, False, 2345, 21, 342 0, False, 2345, 21, 342
0, False, 2561, 244, 532 0, False, 2561, 244, 532

2
tractor.py
View File

@ -5,7 +5,7 @@ from Graph import Graph
from Cucumber import Cucumber from Cucumber import Cucumber
from Tomato import Tomato from Tomato import Tomato
from sklearn import tree from sklearn import tree
from Tree import TreeClass from Tree1 import TreeClass
class Tractor(TreeClass): class Tractor(TreeClass):