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 |
{
|