Emil_Shanaty/NIC_Project
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add: fitness best values

Browse Source
This commit is contained in:
KOSMOGOR
2025-04-14 19:25:08 +03:00
parent ff2ddc98fb
commit bd64901eba
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
genetic_alg.py
+7 -1
View File
@@ -22,6 +22,9 @@ class GeneticAlgorithm:
self.percentageBest = percentageBest self.percentageBest = percentageBest
self.percentageNew = percentageNew self.percentageNew = percentageNew
self.inputSize = inputSize self.inputSize = inputSize
self.fitnessBest = []
self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu") self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(self.device) print(self.device)
self.initialize_population() self.initialize_population()
@@ -53,7 +56,10 @@ class GeneticAlgorithm:
return model return model
def learn(self, fitness: list): def learn(self, fitness: list):
self.population = [self.population[x] for x in np.argsort(fitness)[::-1]] sortedFitnessArg = np.argsort(fitness)[::-1]
self.fitnessBest.append(fitness[sortedFitnessArg[0]])
print(self.fitnessBest[-1])
self.population = [self.population[x] for x in sortedFitnessArg]
numBest = int(self.populationSize * self.percentageBest) numBest = int(self.populationSize * self.percentageBest)
self.population = self.population[:numBest] self.population = self.population[:numBest]
while len(self.population) < self.populationSize - self.populationSize * self.percentageNew: while len(self.population) < self.populationSize - self.populationSize * self.percentageNew: