Upload CodeBERT SQL error classifier
Browse files- config.json +3 -1
- label_config.json +2 -1
- metrics.json +22 -22
- model.safetensors +2 -2
- training_args.bin +1 -1
config.json
CHANGED
|
@@ -19,7 +19,8 @@
|
|
| 19 |
"5": "NULL_HANDLING_ERROR",
|
| 20 |
"6": "PERFORMANCE_ERROR",
|
| 21 |
"7": "LOGICAL_ERROR",
|
| 22 |
-
"8": "SYNTAX_ERROR"
|
|
|
|
| 23 |
},
|
| 24 |
"initializer_range": 0.02,
|
| 25 |
"intermediate_size": 3072,
|
|
@@ -28,6 +29,7 @@
|
|
| 28 |
"FILTER_ERROR": 2,
|
| 29 |
"JOIN_ERROR": 0,
|
| 30 |
"LOGICAL_ERROR": 7,
|
|
|
|
| 31 |
"NULL_HANDLING_ERROR": 5,
|
| 32 |
"PERFORMANCE_ERROR": 6,
|
| 33 |
"SUBQUERY_ERROR": 4,
|
|
|
|
| 19 |
"5": "NULL_HANDLING_ERROR",
|
| 20 |
"6": "PERFORMANCE_ERROR",
|
| 21 |
"7": "LOGICAL_ERROR",
|
| 22 |
+
"8": "SYNTAX_ERROR",
|
| 23 |
+
"9": "NO_ERROR"
|
| 24 |
},
|
| 25 |
"initializer_range": 0.02,
|
| 26 |
"intermediate_size": 3072,
|
|
|
|
| 29 |
"FILTER_ERROR": 2,
|
| 30 |
"JOIN_ERROR": 0,
|
| 31 |
"LOGICAL_ERROR": 7,
|
| 32 |
+
"NO_ERROR": 9,
|
| 33 |
"NULL_HANDLING_ERROR": 5,
|
| 34 |
"PERFORMANCE_ERROR": 6,
|
| 35 |
"SUBQUERY_ERROR": 4,
|
label_config.json
CHANGED
|
@@ -8,7 +8,8 @@
|
|
| 8 |
"NULL_HANDLING_ERROR",
|
| 9 |
"PERFORMANCE_ERROR",
|
| 10 |
"LOGICAL_ERROR",
|
| 11 |
-
"SYNTAX_ERROR"
|
|
|
|
| 12 |
],
|
| 13 |
"model_name": "microsoft/codebert-base",
|
| 14 |
"architecture": "codebert-cross-encoder",
|
|
|
|
| 8 |
"NULL_HANDLING_ERROR",
|
| 9 |
"PERFORMANCE_ERROR",
|
| 10 |
"LOGICAL_ERROR",
|
| 11 |
+
"SYNTAX_ERROR",
|
| 12 |
+
"NO_ERROR"
|
| 13 |
],
|
| 14 |
"model_name": "microsoft/codebert-base",
|
| 15 |
"architecture": "codebert-cross-encoder",
|
metrics.json
CHANGED
|
@@ -1,29 +1,29 @@
|
|
| 1 |
{
|
| 2 |
-
"train_samples":
|
| 3 |
-
"val_samples":
|
| 4 |
-
"test_samples":
|
| 5 |
-
"train_runtime":
|
| 6 |
"validation": {
|
| 7 |
-
"eval_loss": 0.
|
| 8 |
-
"eval_accuracy": 0.
|
| 9 |
-
"eval_f1_macro": 0.
|
| 10 |
-
"eval_f1_micro": 0.
|
| 11 |
-
"eval_precision_macro": 0.
|
| 12 |
-
"eval_recall_macro": 0.
|
| 13 |
-
"eval_hamming_loss": 0.
|
| 14 |
-
"eval_subset_accuracy": 0.
|
| 15 |
-
"eval_runtime":
|
| 16 |
-
"eval_samples_per_second":
|
| 17 |
-
"eval_steps_per_second": 8.
|
| 18 |
"epoch": 2.0
|
| 19 |
},
|
| 20 |
"test": {
|
| 21 |
-
"accuracy": 0.
|
| 22 |
-
"f1_macro": 0.
|
| 23 |
-
"f1_micro": 0.
|
| 24 |
-
"precision_macro": 0.
|
| 25 |
-
"recall_macro": 0.
|
| 26 |
-
"hamming_loss": 0.
|
| 27 |
-
"subset_accuracy": 0.
|
| 28 |
}
|
| 29 |
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"train_samples": 12800,
|
| 3 |
+
"val_samples": 1600,
|
| 4 |
+
"test_samples": 1600,
|
| 5 |
+
"train_runtime": 1720.7174,
|
| 6 |
"validation": {
|
| 7 |
+
"eval_loss": 0.009950438514351845,
|
| 8 |
+
"eval_accuracy": 0.986875,
|
| 9 |
+
"eval_f1_macro": 0.987365893192092,
|
| 10 |
+
"eval_f1_micro": 0.986875,
|
| 11 |
+
"eval_precision_macro": 0.9823432343234325,
|
| 12 |
+
"eval_recall_macro": 0.994,
|
| 13 |
+
"eval_hamming_loss": 0.002625,
|
| 14 |
+
"eval_subset_accuracy": 0.986875,
|
| 15 |
+
"eval_runtime": 24.1234,
|
| 16 |
+
"eval_samples_per_second": 66.326,
|
| 17 |
+
"eval_steps_per_second": 8.291,
|
| 18 |
"epoch": 2.0
|
| 19 |
},
|
| 20 |
"test": {
|
| 21 |
+
"accuracy": 0.988125,
|
| 22 |
+
"f1_macro": 0.9877733307020831,
|
| 23 |
+
"f1_micro": 0.988125,
|
| 24 |
+
"precision_macro": 0.9837556637019634,
|
| 25 |
+
"recall_macro": 0.9932000000000001,
|
| 26 |
+
"hamming_loss": 0.002375,
|
| 27 |
+
"subset_accuracy": 0.988125
|
| 28 |
}
|
| 29 |
}
|
model.safetensors
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:ed2447e6c2792dc691f9c9e72bcf3f530590286c241c81f89163b4ec2d6b3b10
|
| 3 |
+
size 498637432
|
training_args.bin
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 5841
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:be57a5e722508b6c76ce1a40aa31a271c24b48cfd904164015ebde41dfb08a47
|
| 3 |
size 5841
|