CoolFace
Modelpublic

KakashiH/Image_Based_Malware_Classification_ViT

sourceHugging Faceapache-2.0updated 2y agoView on Hugging Face
1likes
Model Card

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