KakashiH's picture
Update README.md
82632ce verified
metadata
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