From 9fc28b3d630a22c137fa2e8792b04cea3d083bf1 Mon Sep 17 00:00:00 2001
From: Sun Jin Kim <sk2521@ic.ac.uk>
Date: Wed, 27 Apr 2022 17:14:36 +0100
Subject: [PATCH] privatize further mtehods in evo_learner
---
MetaAugment/autoaugment_learners/evo_learner.py | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/MetaAugment/autoaugment_learners/evo_learner.py b/MetaAugment/autoaugment_learners/evo_learner.py
index a7dc580d..f0547b91 100644
--- a/MetaAugment/autoaugment_learners/evo_learner.py
+++ b/MetaAugment/autoaugment_learners/evo_learner.py
@@ -194,11 +194,11 @@ class evo_learner(aa_learner):
def _set_up_instance(self, train_dataset, test_dataset, child_network_architecture):
"""
- Initialises GA instance, as well as fitness and on_generation functions
+ Initialises GA instance, as well as fitness and _on_generation functions
"""
- def fitness_func(solution, sol_idx):
+ def _fitness_func(solution, sol_idx):
"""
Defines the fitness function for the parent selection
@@ -252,7 +252,7 @@ class evo_learner(aa_learner):
return fit_val
- def on_generation(ga_instance):
+ def _on_generation(ga_instance):
"""
Prints information of generational fitness
@@ -273,5 +273,5 @@ class evo_learner(aa_learner):
num_parents_mating=self.num_parents_mating,
initial_population=self.initial_population,
mutation_percent_genes = 0.1,
- fitness_func=fitness_func,
- on_generation = on_generation)
+ _fitness_func=_fitness_func,
+ _on_generation = _on_generation)
--
GitLab