Skip to main content

Table 2 GA code snippet for defining Optimal CNN structure

From: Medical image diagnosis based on adaptive Hybrid Quantum CNN

def fitness(pop, X, y, epochs):

pop_accuracy = []

for i in range(len(pop)):

num_layers = pop[i][0]

n_filters = pop[i][1:1 + num_layers]

s_filters = pop[i][1 + num_layers:]

model = cnn_model(num_layers, n_filters, s_filters)

k = model.fit(X, y, batch_size = 32, epochs = epochs)

accuracy = k.history["accuracy"]

pop_accuracy.append(max(accuracy))

return pop_accuracy