This is a model that detects whether a pe file as image is malicious or not.
You can access the dataset used to train the model @: https://www.kaggle.com/datasets/ang3loliveira/malware-analysis-datasets-raw-pe-as-image
Results from last Epoch:
Epoch 8, Loss: 0.0274, Val Acc: 0.9782
precision recall f1-score support
0 0.96 1.00 0.98 9860
1 1.00 0.96 0.98 9891
accuracy 0.98 19751
macro-avg 0.98 0.98 0.98 19751
weighted-avg 0.98 0.98 0.98 19751