Gotowy podprojekt

This commit is contained in:
xkamikoo 2020-05-18 17:13:09 +02:00
parent 3666bd3079
commit 89ac7e6da6

View File

@ -4,61 +4,34 @@ from sklearn.model_selection import train_test_split
from sklearn import metrics
import numpy
header = ["ready", "hydration", "weeds", "empty", "TODO"]
work = ["Zebrac","Podlac","Odchwascic","Zasadzic"]
#0 - 3
#1 - 0
#2 - 1
#3 - 2
def check_p(field):
if field == 0:
return [0, 0, 0, 0, "Zasadzic"]
elif field == 1:
return [0, 0, 1, 0, "Odchwascic"]
elif field == 2:
return [0, 0, 0, 1, "Podlac"]
elif field == 3:
return [0, 0, 1, 1, "Odchwascic"]
elif field == 4:
return [0, 1, 0, 0, "Zasadzic"]
elif field == 5:
return [0, 1, 1, 0, "Odchwascic"]
elif field == 6:
return [0, 1, 0, 1, "Ignoruj"]
elif field == 7:
return [0, 1, 1, 1, "Odchwascic"]
elif field == 8:
return [1, 0, 0, 1, "Zebrac"]
else:
print("wrong field number")
header = ["hydration", "weeds", "empty", "ready", "TODO"]
work = ["Podlac", "Odchwascic", "Zasadzic", "Zebrac"]
def check(field):
if field == 0:
return [[0, 0, 0, 1, "Zasadzic"],[0,0,0,1,"Podlac"]]
return [[0, 0, 1, 0, "Zasadzic"], [0, 0, 1, 0, "Podlac"]]
elif field == 1:
return [[0, 0, 1, 1, "Odchwascic"], [0,0,1,1,"Podlac"], [0,0,1,1,"Zasadzic"]]
return [[0, 1, 1, 0, "Odchwascic"], [0, 1, 1, 0, "Podlac"], [0, 1, 1, 0, "Zasadzic"]]
elif field == 2:
return [[0, 0, 0, 0, "Podlac"]]
elif field == 3:
return [[0, 0, 1, 0, "Odchwascic"],[0,0,1,0,"Podlac"]]
return [[0, 1, 0, 0, "Odchwascic"], [0, 1, 0, 0, "Podlac"]]
elif field == 4:
return [[0, 1, 0, 1, "Zasadzic"]]
return [[1, 0, 1, 0, "Zasadzic"]]
elif field == 5:
return [[0, 1, 1, 1, "Odchwascic"],[0,1,1,1,"Zasadzic"]]
return [[1, 1, 1, 0, "Odchwascic"], [1, 1, 1, 0, "Zasadzic"]]
elif field == 6:
return []
elif field == 7:
return [[0, 1, 1, 0, "Odchwascic"]]
return [[1, 1, 0, 0, "Odchwascic"]]
elif field == 8:
return [[1, 0, 0, 0, "Zebrac"],[1, 0, 0, 0, "Potem podlac"],[1, 0, 0, 0, "Potem zasadzic"]]
return [[0, 0, 0, 1, "Zebrac"], [0, 0, 0, 1, "Potem podlac"], [0, 0, 0, 1, "Potem zasadzic"]]
else:
print("wrong field number")
def un_values(rows, col):
return set([row[col] for row in rows])
# liczenie ilości prac do wykonania
def class_counts(rows):
counts = {}
for row in rows:
@ -69,10 +42,12 @@ def class_counts(rows):
return counts
# sprawdzenie czy wartość jest liczbą
def is_numeric(value):
return isinstance(value, int) or isinstance(value, float)
# klasa tworząca zapytanie do podziału danych
class Question():
def __init__(self, column, value):
self.column = column
@ -82,11 +57,9 @@ class Question():
val = example[self.column]
if is_numeric(val):
return val == self.value
else:
return val != self.value
# wyświetlenie pytania
def __repr__(self):
condition = "!="
if is_numeric(self.value):
condition = "=="
return "Is %s %s %s?" % (
@ -94,6 +67,7 @@ class Question():
)
# podział danych na spełnione i niespełnione wiersze
def partition(rows, question):
true_rows, false_rows = [], []
for row in rows:
@ -104,6 +78,7 @@ def partition(rows, question):
return true_rows, false_rows
# funkcja implementująca indeks gini
def gini(rows):
counts = class_counts(rows)
impurity = 1
@ -118,6 +93,7 @@ def info_gain(left, right, current_uncertainty):
return current_uncertainty - p * gini(left) - (1 - p) * gini(right)
# znalezienie najlepszego "miejsca" na podział danych
def find_best_split(rows):
best_gain = 0
best_question = None
@ -152,6 +128,7 @@ class DecisionNode:
self.false_branch = false_branch
# funkcja budująca drzewo
def build_tree(rows):
gain, question = find_best_split(rows)
if gain == 0:
@ -164,6 +141,7 @@ def build_tree(rows):
return DecisionNode(question, true_branch, false_branch)
# funcka wypisująca drzewo
def print_tree(node, spacing=""):
if isinstance(node, Leaf):
print(spacing + "Predict", node.predictions)
@ -178,23 +156,6 @@ def print_tree(node, spacing=""):
print_tree(node.false_branch, spacing + " ")
def classify(row, node):
if isinstance(node, Leaf):
return node.predictions
if node.question.match(row):
return classify(row, node.true_branch)
else:
return classify(row,node.false_branch)
def print_leaf(counts):
total = sum(counts.values()) * 1.0
probs = {}
for lbl in counts.keys():
probs[lbl] = str(int(counts[lbl]/total * 100)) + "%"
return probs
class main():
def __init__(self, traktor, field, ui, path):
self.traktor = traktor
@ -202,7 +163,9 @@ class main():
self.ui = ui
self.path = path
self.best_action = 0
def main(self):
# dane testowe
array = ([[8, 8, 8, 8, 8, 8, 8, 8, 8, 8],
[7, 7, 7, 7, 7, 7, 7, 7, 7, 7],
[6, 6, 6, 6, 6, 6, 6, 6, 6, 6],
@ -213,12 +176,14 @@ class main():
[1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0]])
while (self.best_action != -1):
while (True):
self.find_best_action()
if self.best_action == -1:
break
self.do_best_action()
print("Koniec roboty")
def find_best_action(self):
testing_data = []
matrix = self.field.get_matrix()
@ -247,20 +212,13 @@ class main():
else:
self.best_action = self.find_remaining_action(matrix_todo)
return
#for row in testing_data:
# print("Actual: %s. Predicted %s" %
# (row[-1], print_leaf(classify(row, x))))
#for row in matrix_todo:
# print(row)
def do_best_action(self):
self.traktor.set_mode((self.best_action+3) % 4)
self.traktor.set_mode(self.best_action)
while self.path.pathfinding(self.traktor, self.field, self.ui) != 0:
pass
# 0 - 3
# 1 - 0
# 2 - 1
# 3 - 2
def find_remaining_action(self, matrix_todo):
for row in matrix_todo:
for field in row: