Instructions to use endorlabs/malicious-package-classifier-bert-mal-only with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use endorlabs/malicious-package-classifier-bert-mal-only with PEFT:
from peft import PeftModel from transformers import AutoModelForSequenceClassification base_model = AutoModelForSequenceClassification.from_pretrained("microsoft/codebert-base-mlm") model = PeftModel.from_pretrained(base_model, "endorlabs/malicious-package-classifier-bert-mal-only") - Notebooks
- Google Colab
- Kaggle
End of training
Browse files
README.md
CHANGED
|
@@ -15,7 +15,7 @@ should probably proofread and complete it, then remove this comment. -->
|
|
| 15 |
|
| 16 |
This model is a fine-tuned version of [microsoft/codebert-base-mlm](https://huggingface.co/microsoft/codebert-base-mlm) on an unknown dataset.
|
| 17 |
It achieves the following results on the evaluation set:
|
| 18 |
-
- Loss: 0.
|
| 19 |
|
| 20 |
## Model description
|
| 21 |
|
|
@@ -40,44 +40,15 @@ The following hyperparameters were used during training:
|
|
| 40 |
- seed: 42
|
| 41 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
| 42 |
- lr_scheduler_type: linear
|
| 43 |
-
- num_epochs:
|
| 44 |
|
| 45 |
### Training results
|
| 46 |
|
| 47 |
| Training Loss | Epoch | Step | Validation Loss |
|
| 48 |
|:-------------:|:-----:|:----:|:---------------:|
|
| 49 |
-
| No log | 1.0 | 42 | 0.
|
| 50 |
-
| No log | 2.0 | 84 | 0.
|
| 51 |
-
| No log | 3.0 | 126 | 0.
|
| 52 |
-
| No log | 4.0 | 168 | 0.0005 |
|
| 53 |
-
| No log | 5.0 | 210 | 0.0003 |
|
| 54 |
-
| No log | 6.0 | 252 | 0.0003 |
|
| 55 |
-
| No log | 7.0 | 294 | 0.0002 |
|
| 56 |
-
| No log | 8.0 | 336 | 0.0002 |
|
| 57 |
-
| No log | 9.0 | 378 | 0.0001 |
|
| 58 |
-
| No log | 10.0 | 420 | 0.0001 |
|
| 59 |
-
| No log | 11.0 | 462 | 0.0001 |
|
| 60 |
-
| 0.0161 | 12.0 | 504 | 0.0001 |
|
| 61 |
-
| 0.0161 | 13.0 | 546 | 0.0001 |
|
| 62 |
-
| 0.0161 | 14.0 | 588 | 0.0001 |
|
| 63 |
-
| 0.0161 | 15.0 | 630 | 0.0001 |
|
| 64 |
-
| 0.0161 | 16.0 | 672 | 0.0001 |
|
| 65 |
-
| 0.0161 | 17.0 | 714 | 0.0001 |
|
| 66 |
-
| 0.0161 | 18.0 | 756 | 0.0001 |
|
| 67 |
-
| 0.0161 | 19.0 | 798 | 0.0000 |
|
| 68 |
-
| 0.0161 | 20.0 | 840 | 0.0000 |
|
| 69 |
-
| 0.0161 | 21.0 | 882 | 0.0000 |
|
| 70 |
-
| 0.0161 | 22.0 | 924 | 0.0000 |
|
| 71 |
-
| 0.0161 | 23.0 | 966 | 0.0000 |
|
| 72 |
-
| 0.0 | 24.0 | 1008 | 0.0000 |
|
| 73 |
-
| 0.0 | 25.0 | 1050 | 0.0000 |
|
| 74 |
-
| 0.0 | 26.0 | 1092 | 0.0000 |
|
| 75 |
-
| 0.0 | 27.0 | 1134 | 0.0000 |
|
| 76 |
-
| 0.0 | 28.0 | 1176 | 0.0000 |
|
| 77 |
-
| 0.0 | 29.0 | 1218 | 0.0000 |
|
| 78 |
-
| 0.0 | 30.0 | 1260 | 0.0000 |
|
| 79 |
-
| 0.0 | 31.0 | 1302 | 0.0000 |
|
| 80 |
-
| 0.0 | 32.0 | 1344 | 0.0000 |
|
| 81 |
|
| 82 |
|
| 83 |
### Framework versions
|
|
|
|
| 15 |
|
| 16 |
This model is a fine-tuned version of [microsoft/codebert-base-mlm](https://huggingface.co/microsoft/codebert-base-mlm) on an unknown dataset.
|
| 17 |
It achieves the following results on the evaluation set:
|
| 18 |
+
- Loss: 0.0026
|
| 19 |
|
| 20 |
## Model description
|
| 21 |
|
|
|
|
| 40 |
- seed: 42
|
| 41 |
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
| 42 |
- lr_scheduler_type: linear
|
| 43 |
+
- num_epochs: 3
|
| 44 |
|
| 45 |
### Training results
|
| 46 |
|
| 47 |
| Training Loss | Epoch | Step | Validation Loss |
|
| 48 |
|:-------------:|:-----:|:----:|:---------------:|
|
| 49 |
+
| No log | 1.0 | 42 | 0.0104 |
|
| 50 |
+
| No log | 2.0 | 84 | 0.0034 |
|
| 51 |
+
| No log | 3.0 | 126 | 0.0026 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
|
| 53 |
|
| 54 |
### Framework versions
|
adapter_model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 3555504
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:37f5e2f76086f8ad526ab9eaf1ee10dd314d97d3c1b435566404f5f76ddeea94
|
| 3 |
size 3555504
|
logs/events.out.tfevents.1717318151.llms-a100.332532.1
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:418849d2e349e65071bf1469deaae0dddaea51a69c4c0df6f30ebd74e75c6a2d
|
| 3 |
+
size 6075
|
logs/events.out.tfevents.1717318438.llms-a100.332532.2
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:da46fdde3cc68331aa5caec00a5426d0f9f6201504496172034a9c498d21a0d2
|
| 3 |
+
size 354
|