train.py

#Change the parameters to whatever suits you
batch_size = 512 
epochs = 100
labels = [0 for _ in benign_images] + [1 for _ in malicious_images]
model.fit(benign_images+malicious_images, labels,
              batch_size = batch_size,
              epochs = epochs,
              validation_split = 0.25,
              shuffle = True)

Hello Islem,

I try this code, but the benign_images and malicious_images aren't defined, and couldn't see the gist that you mention above. Can you please provide the full script?