---
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: 

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