kuhyx/WUT_Computer_Science
1
0
Fork 0
mirror of https://github.com/kuhyx/WUT_Computer_Science.git synced 2026-07-04 17:43:12 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: add counting generating time

Browse Source
This commit is contained in:
Krzysztof Rudnicki 2023-04-12 19:21:12 +02:00
parent d19db606ea
commit 10b5a511ca
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
lab3/main.py
View File

@ -4,6 +4,7 @@ Program that optimizes Rastrigin function: f (x, y) =
Using Evolutionary Strategy (μ, λ).
"""
import sys
import time
import numpy as np
@ -51,7 +52,7 @@ def evolution_strategy(
low=min_max[0], high=min_max[1], size=(
size_of_population, 2))
# Iterate untill we reach max number of generate and terminate
# Iterate untill we reach max number of generate and terminate
for generation_number in range(number_of_generations):
fitness, population = generate(
generation_number,
@ -129,12 +130,19 @@ def user_input():
if __name__ == "__main__":
# Run the Evolutionary Strategy algorithm
ARGUMENTS = user_input()
start_time = time.perf_counter()
best_individual, best_fitness = evolution_strategy(
ARGUMENTS["number_of_parents"],
ARGUMENTS["size_of_population"],
ARGUMENTS["mutation_strength"],
ARGUMENTS["number_of_generations"],
(ARGUMENTS["min"], ARGUMENTS["max"]))
end_time = time.perf_counter()
total_generation_time = end_time - start_time
time_per_generation = total_generation_time / \
ARGUMENTS["number_of_generations"]
print("Best individual found:", best_individual)
print("Best fitness found:", best_fitness)
print("total_generation_time: ", total_generation_time)
print("time_per_generation: ", time_per_generation)