Update README.md

README.md

@@ -3,17 +3,20 @@ tags:
 - generated_from_trainer
 metrics:
 - accuracy
+- f1
+- recall
+- precision
 model-index:
 - name: codebert-base-Malicious_URLs
   results: []
+language:
+- en
+pipeline_tag: text-classification
 ---
 
-<!-- This model card has been generated automatically according to the information the Trainer had access to. You
-should probably proofread and complete it, then remove this comment. -->
-
 # codebert-base-Malicious_URLs
 
-This model is a fine-tuned version of [microsoft/codebert-base](https://huggingface.co/microsoft/codebert-base) on the None dataset.
+This model is a fine-tuned version of [microsoft/codebert-base](https://huggingface.co/microsoft/codebert-base).
 It achieves the following results on the evaluation set:
 - Loss: 0.8225
 - Accuracy: 0.7279
@@ -29,15 +32,15 @@ It achieves the following results on the evaluation set:
 
 ## Model description
 
-More information needed
+For more information on how it was created, check out the following link: https://github.com/DunnBC22/NLP_Projects/blob/main/Multiclass%20Classification/Malicious%20URLs/Malicious%20URLs%20-%20CodeBERT.ipynb
 
 ## Intended uses & limitations
 
-More information needed
+This model is intended to demonstrate my ability to solve a complex problem using technology.
 
 ## Training and evaluation data
 
-More information needed
+Dataset Source: https://www.kaggle.com/datasets/sid321axn/malicious-urls-dataset
 
 ## Training procedure
 
@@ -64,4 +67,4 @@ The following hyperparameters were used during training:
 - Transformers 4.27.4
 - Pytorch 2.0.0
 - Datasets 2.11.0
-- Tokenizers 0.13.3
+- Tokenizers 0.13.3