Sleoruiz
/

distilbert-fine-tuned-text-classification-SL-data-augmentation

Text Classification

Model card Files Files and versions

Sleoruiz commited on May 2, 2023

Commit

627d40b

·

1 Parent(s): 8f1917e

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -139,7 +139,7 @@
   "n_layers": 6,
   "output_past": true,
   "pad_token_id": 0,
-  "problem_type": "multi_class_classification",
   "qa_dropout": 0.1,
   "seq_classif_dropout": 0.2,
   "sinusoidal_pos_embds": false,

   "n_layers": 6,
   "output_past": true,
   "pad_token_id": 0,
+  "problem_type": "single_label_classification",
   "qa_dropout": 0.1,
   "seq_classif_dropout": 0.2,
   "sinusoidal_pos_embds": false,