stochastiscm_ga_card.py

def mutate(dna, mutation_rate):
    for i in range(len(dna)):
        if np.random.random_sample() < mutation_rate:
            ri = np.random.randint(0, 10)
            while ri == i:
                ri = np.random.randint(0, 10)
            dna[i], dna[ri] = dna[ri], dna[i] 
    return dna


def crossover(dna1, dna2):
    cross_index = np.random.randint(1, 9)
    new_dna1 = dna1[:cross_index].tolist()
    new_dna2 = [d for d in dna2 if d not in new_dna1]
    new_dna = np.array(new_dna1 + new_dna2)
    assert len(new_dna) == len(dna1)
    assert len(np.unique(new_dna)) == len(np.unique(dna2))
    return new_dna