75 lines
1.8 KiB
Python
Executable File
75 lines
1.8 KiB
Python
Executable File
#!/usr/bin/env python
|
|
import math
|
|
import pygame
|
|
import random
|
|
import time
|
|
|
|
from car import Car
|
|
from genetics import genetic_selection, genetic_reproduction
|
|
from maps import map1
|
|
from params import CELL_COLOR, screen
|
|
|
|
|
|
|
|
#https://medium.com/intel-student-ambassadors/demystifying-genetic-algorithms-to-enhance-neural-networks-cde902384b6e
|
|
clock = pygame.time.Clock()
|
|
|
|
|
|
map_lines = map1
|
|
|
|
|
|
|
|
all_cars = pygame.sprite.Group()
|
|
|
|
for x in range(100):
|
|
car = Car()
|
|
all_cars.add(car)
|
|
|
|
|
|
def run_round(all_cars):
|
|
running_cars = True
|
|
while running_cars :
|
|
running_cars = False
|
|
screen.fill(CELL_COLOR)
|
|
all_cars.draw(screen)
|
|
for c in all_cars :
|
|
c.show_features()
|
|
if c.run :
|
|
running_cars = True
|
|
c.probe_lines_proximity(map_lines)
|
|
c.probe_brain()
|
|
c.update()
|
|
|
|
for line in map_lines :
|
|
pygame.draw.line(screen, (255,255,255), line[0], line[1])
|
|
|
|
pygame.display.flip()
|
|
clock.tick(48)
|
|
|
|
# for c in all_cars :
|
|
# print(f"Car {id(c)} Fitness : {c.brain.fitness})")
|
|
|
|
print('Collecting brains')
|
|
brains = [c.brain for c in all_cars]
|
|
print(f"Max fitness = {max([b.fitness for b in brains])}" )
|
|
print(f"Avg fitness = {sum([b.fitness for b in brains])/len(brains)}" )
|
|
print('selecting')
|
|
parents_pool = genetic_selection(brains)
|
|
# import ipdb; ipdb.set_trace()
|
|
print("breeding")
|
|
new_brains = genetic_reproduction(parents_pool)
|
|
print(f'building {len(new_brains)} cars with new brains')
|
|
all_cars.empty()
|
|
for b in new_brains :
|
|
all_cars.add(Car(brain=b))
|
|
print("Waiting before new run")
|
|
for x in range(1) :
|
|
time.sleep(0.5)
|
|
pygame.display.flip()
|
|
|
|
|
|
while True :
|
|
run_round(all_cars)
|
|
pygame.display.flip()
|
|
clock.tick(24)
|