| --- |
| license: apache-2.0 |
| language: |
| - en |
| metrics: |
| - accuracy |
| base_model: |
| - google/vit-base-patch16-224 |
| pipeline_tag: image-classification |
| library_name: peft |
| tags: |
| - Malware |
| - CyberSecurity |
| - MalwareClassification |
| --- |
| |
| ## Finetuned google/vit-base-patch16-224 on Microsoft Malware Data (BIG 2015) for iamge based malware classification. |
|
|
|
|
| ## Data Source: https://www.kaggle.com/c/malware-classification/data |
|
|
|
|
| ## logs : https://wandb.ai/prasad-cdac-work/vit-malware-classification-microsoft?nw=nwuserprasadcdacwork |
|
|
| ## Classification Report: |
|
|
|  |
|
|
