individual.py

import predictor
from digit_mutator import DigitMutator
import numpy as np

class Individual(object):
    # Global counter of all the individuals (it is increased each time an individual is created or mutated).
    COUNT = 0
    SEEDS = set()
    COUNT_MISS = 0

    def __init__(self, member, seed):
        self.seed = seed
        self.ff = None
        self.member = member
        self.features = tuple()

    def reset(self):
        self.ff = None

    def evaluate(self):
        if self.ff is None:          
            self.member.predicted_label, self.member.confidence = \
                predictor.Predictor.predict(self.member.purified)

            # Calculate fitness function
            self.ff = self.member.confidence if self.member.confidence > 0 else -0.1
            
        return self.ff

    def mutate(self):
        DigitMutator(self.member).mutate()
        self.reset()