endorlabs
/

malicious-package-classifier-bert-mal-only

Generated from Trainer

Model card Files Files and versions Metrics Training metrics Community

schnapp commited on May 31

Commit

cf598aa

•

1 Parent(s): 3dae9f2

End of training

Files changed (1) hide show

config.json +1 -0

config.json CHANGED Viewed

@@ -26,6 +26,7 @@
   "output_past": true,
   "pad_token_id": 1,
   "position_embedding_type": "absolute",
   "transformers_version": "4.39.3",
   "type_vocab_size": 1,
   "use_cache": true,

   "output_past": true,
   "pad_token_id": 1,
   "position_embedding_type": "absolute",
+  "problem_type": "single_label_classification",
   "transformers_version": "4.39.3",
   "type_vocab_size": 1,
   "use_cache": true,