File size: 717 Bytes
1ae6df5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
237c3af
1ae6df5
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
---
license: apache-2.0
language:
- en
metrics:
- accuracy
base_model:
- google/vit-base-patch16-224
pipeline_tag: image-classification
library_name: peft
tags:
- CyberSecurity
- MalwareImageClassification
- Malware
---

## Finetuned google/vit-base-patch16-224 on Malimg Dataset contains 9,339 malware byteplot images from 25 different families. For iamge based malware classification.


## Data Source: https://www.kaggle.com/datasets/ikrambenabd/malimg-original


## logs : https://wandb.ai/prasad-cdac-work/vit-malware-classification-MalImg?nw=nwuserprasadcdacwork


## Classification Report:

![image/png](https://cdn-uploads.huggingface.co/production/uploads/65deccec32ce1e0dd304e025/55QlZCtATNYdb44JHbrez.png)