manueldeprada
/

ctTREC-distillbert-correct-classifier-trec2020

Text Classification

Model card Files Files and versions

Manuel de Prada commited on Apr 18, 2022

Commit

e539347

·

1 Parent(s): 7914e70

add model

Files changed (2) hide show

config.json +3 -3
flax_model.msgpack +3 -0

config.json CHANGED Viewed

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "cesga_run_12_4.train_2020/correctness-classifier_batch16/checkpoint-500",
   "activation": "gelu",
   "architectures": [
     "DistilBertForSequenceClassification"
@@ -10,8 +10,8 @@
   "finetuning_task": "sst-2",
   "hidden_dim": 3072,
   "id2label": {
-    "0": "incorrect",
-    "1": "correct"
   },
   "initializer_range": 0.02,
   "label2id": {

 {
+  "_name_or_path": "distilbert-base-uncased-finetuned-sst-2-english",
   "activation": "gelu",
   "architectures": [
     "DistilBertForSequenceClassification"
   "finetuning_task": "sst-2",
   "hidden_dim": 3072,
   "id2label": {
+    "0": "NEGATIVE",
+    "1": "POSITIVE"
   },
   "initializer_range": 0.02,
   "label2id": {

flax_model.msgpack ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:4d4d38d65d15b22e24d220a204b4f07d4982c3c67fd04a7ec5f805f3802cf2e8
+size 267823701