遗传算法实例：句子匹配 python实现

kalifa_lau

发布于 2019-04-01 14:58:00

1.2K0

发布于 2019-04-01 14:58:00

文章被收录于专栏：kalifaの日々kalifaの日々

题目来自莫烦python教学

tips:

1）当你的算法总是不收敛，诶反正就是你怎么改参数它都不收敛的时候，可能是fitness函数写错了(幽怨脸)，问问自己，numpy矩阵操作对了吗？打个输出看看真的符合预期吗？

2）把numpy数组里的数字按照ascii编码变成字符串：

row = np.array([123,122,98]).astype(np.int8) #一定一定要astype(np.int8)否则会出错
row = row.tostring().decode("ascii")

代码实现效果：

GIF.gif

源代码：

import numpy as np

TARGET = 'Do you ever loved me' #雾草我居然用这么酸的话当输入？
DNA_SIZE = len(TARGET)
GENERATION = 10000
CROSSOVER_RATE = 0.4
MUTATE_RATE = 0.01
POP_SIZE = 300
DNA_BOUND = [32,123]
TARGET_ARR = np.fromstring(TARGET,dtype = np.uint8)

class GA(object):
    def __init__(self):
        self.pop = np.random.randint(DNA_BOUND[0],DNA_BOUND[1],(1,DNA_SIZE)).astype(np.int8).repeat(POP_SIZE,axis=0)
    
    def getFitness(self,pop):
        root = np.fromstring(TARGET,dtype = np.int8).reshape(1,DNA_SIZE).repeat(POP_SIZE,axis=0)
        #print("root is ",root)
        root = root - pop
        return np.sum(root==0,axis=1)
        
        
    def select(self,fitness):
        
        idx = np.random.choice(np.arange(POP_SIZE),size = POP_SIZE,replace = True,p = fitness/fitness.sum())
        #print("idx : ",idx)
        
        return self.pop[idx]
    
    def mutate(self,child):
        for index in range(DNA_SIZE):
            if np.random.rand() < MUTATE_RATE:
                child[index] = np.random.randint(DNA_BOUND[0],DNA_BOUND[1],size=1)
        return child
    
    def crossover(self,parent,pop):
        if np.random.rand() < CROSSOVER_RATE:
            i = np.random.randint(0,POP_SIZE,size=1)
            
            cross_points = np.random.randint(0,2,size=DNA_SIZE).astype(np.bool)
            parent[cross_points] = pop[i,cross_points]
        return parent
            
    def translateDNA(self,row):
        #print("row:",row,"length ",len(row))

        return row.tostring().decode('ascii')
    
    def evolution(self,gen):
        fitness = self.getFitness(self.pop) + 1e-4
        
        self.pop = self.select(fitness)
        
        #print("Gen : ",gen,"pop :",self.pop)
        bestRes = self.translateDNA(self.pop[np.argmax(fitness)])

        print("Gen : ",gen,"best result:",bestRes," target is ",TARGET)
        
        pop_copy = self.pop.copy()
        
        for parent in self.pop:
            child = self.crossover(parent,pop_copy)
            child = self.mutate(child)
            parent[:] = child
            
        return bestRes
    
    
if __name__ == '__main__':
    
    a = GA()
    for gen in range(GENERATION):
        res = a.evolution(gen)
        if res == TARGET :
            break