s474139/Traktor
2
1
Fork 0
Code Issues Pull Requests 1 Packages Projects Releases Wiki Activity

Update main.py

Browse Source
This commit is contained in:
s481846 2024-05-23 02:14:06 +02:00
parent 268a39aa2d
commit 7091a99ee6
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
main.py
View File

@ -5,13 +5,15 @@ from tractor import Tractor
from kolejka import Stan, Odwiedzone from kolejka import Stan, Odwiedzone
from queue import Queue from queue import Queue
import pandas as pd import pandas as pd
from neuralnetwork import load_model
data = pd.read_csv('dane.csv') data = pd.read_csv('dane.csv')
from decisiontree import train_decision_tree from decisiontree import train_decision_tree
model_path = 'model.pth'
neuralnetwork_model = load_model(model_path)
model, feature_columns = train_decision_tree(data) model, feature_columns = train_decision_tree(data)
@ -60,7 +62,7 @@ def main():
clock = pygame.time.Clock() clock = pygame.time.Clock()
board = Board() board = Board()
board.load_images() board.load_images()
tractor = Tractor(4, 4, model, feature_columns) tractor = Tractor(4, 4, model, feature_columns, neuralnetwork_model)
while run: while run:
clock.tick(fps) clock.tick(fps)
@ -75,7 +77,7 @@ def main():
print("Nie znaleziono ścieżki do najbliższego pola dirt.") print("Nie znaleziono ścieżki do najbliższego pola dirt.")
board.generate_board() board.generate_board()
initial_state = Stan(4, 4, "down") initial_state = Stan(4, 4, "down")
tractor = Tractor(4, 4, model, feature_columns) tractor = Tractor(4, 4, model, feature_columns, neuralnetwork_model)
while board.is_rock(initial_state.row, initial_state.col): while board.is_rock(initial_state.row, initial_state.col):
board.generate_board() board.generate_board()
continue continue