Update README.md
Browse files
README.md
CHANGED
|
@@ -9,7 +9,7 @@ pipeline_tag: text-classification
|
|
| 9 |
|
| 10 |
# DroidDetect-Large
|
| 11 |
|
| 12 |
-
This is a text classification model based on `answerdotai/ModernBERT-large`, fine-tuned to distinguish between **human-written**,
|
| 13 |
|
| 14 |
The model was trained on the `DroidCollection` dataset. It's designed as a **4-class classifier** to address the core task of AI code detection.
|
| 15 |
|
|
@@ -25,7 +25,7 @@ A key feature of this model is its training objective, which combines standard *
|
|
| 25 |
|
| 26 |
#### Label Mapping
|
| 27 |
|
| 28 |
-
The model predicts one of
|
| 29 |
|
| 30 |
```json
|
| 31 |
{
|
|
|
|
| 9 |
|
| 10 |
# DroidDetect-Large
|
| 11 |
|
| 12 |
+
This is a text classification model based on `answerdotai/ModernBERT-large`, fine-tuned to distinguish between **human-written**, and **AI-generated** code.
|
| 13 |
|
| 14 |
The model was trained on the `DroidCollection` dataset. It's designed as a **4-class classifier** to address the core task of AI code detection.
|
| 15 |
|
|
|
|
| 25 |
|
| 26 |
#### Label Mapping
|
| 27 |
|
| 28 |
+
The model predicts one of 2 classes. The mapping from ID to label is as follows:
|
| 29 |
|
| 30 |
```json
|
| 31 |
{
|