diff --git a/car.py b/car.py index a1db562..d357ede 100644 --- a/car.py +++ b/car.py @@ -1,3 +1,4 @@ +import math import numpy as np import random import pygame @@ -47,6 +48,7 @@ class Car(pygame.sprite.Sprite): self.update_sensors() self.probe_brain() self.run = True + self.distance_run = 0 def reset_car_pos(self): self.rect.center = ( @@ -74,7 +76,8 @@ class Car(pygame.sprite.Sprite): old_center = self.rect.center self.rect.center = (self.speed * vec[0] / 2 + old_center[0], -self.speed * vec[1] / 2 + old_center[1]) self.update_sensors() - self.brain.fitness += int(distance(old_center, self.rect.center)) + self.distance_run += int(distance(old_center, self.rect.center)) + self.brain.fitness = math.sqrt(self.distance_run)