ShahzadSohail commited on
Commit
b282ea8
·
verified ·
1 Parent(s): 214af31

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +6 -0
config.json CHANGED
@@ -1,4 +1,10 @@
1
  {
 
 
 
 
 
 
2
  "architectures": [
3
  "RobertaForSequenceClassification"
4
  ],
 
1
  {
2
+ "task_specific_params": {
3
+ "text-classification": {
4
+ "return_all_scores": true
5
+ }
6
+ }
7
+
8
  "architectures": [
9
  "RobertaForSequenceClassification"
10
  ],