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 |