URLBERT Tiny v3 Malicious URL Classifier This is a lightweight version of BERT, specifically fine tuned for classifying URLs into four categories: benign, phishing, malware, and defacement. Model Details Model size : 3.69M parameters Tensor type : F32 Model weight size : 14.8 MB Base model : CrabInHoney/urlbert tiny base v3 Dataset : Malicious URLs Dataset Model Evaluation Results The model was evaluated on a test set with the following classification metrics: Class Precision Recall F1 Score Benign 0.987695 0.993717 0.990697 Defacement 0.988510 0.998963 0.993709 Malware 0.988291 0.960332 0.974111 Phishing 0.958425 0.930826 0.944423 Accuracy 0.983738 0.983738 0.983738 Macro Avg 0.980730 0.970959 0.975735 Weighted Avg 0.983615 0.983738 0.983627 Usage Example Below is an example of how to use the model for URL classification using the Hugging Face transformers library: Example Output:
We use cookies for essential functionality and analytics. You can accept or reject analytics cookies.Cookie policy