Hugging Face's logo

tomaarsen
/
reranker-NeoBERT-gooaq-bce

Text Ranking
sentence-transformers
Safetensors
English
neobert
cross-encoder
text-classification
Generated from Trainer
dataset_size:578402
loss:BinaryCrossEntropyLoss
custom_code
Model card Files
xet
Community
tomaarsen HF Staff commited on
Commit
330dab1
·
verified ·
1 Parent(s): d7354d1

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +1 -1
config.json CHANGED
@@ -7,7 +7,7 @@
7
  "AutoConfig": "model.NeoBERTConfig",
8
  "AutoModel": "model.NeoBERTLMHead",
9
  "AutoModelForMaskedLM": "model.NeoBERTLMHead",
10
- "AutoModelForSequenceClassification": "chandar-lab/NeoBERT--model.NeoBERTForSequenceClassification"
11
  },
12
  "classifier_init_range": 0.02,
13
  "decoder_init_range": 0.02,
 
7
  "AutoConfig": "model.NeoBERTConfig",
8
  "AutoModel": "model.NeoBERTLMHead",
9
  "AutoModelForMaskedLM": "model.NeoBERTLMHead",
10
+ "AutoModelForSequenceClassification": "model.NeoBERTForSequenceClassification"
11
  },
12
  "classifier_init_range": 0.02,
13
  "decoder_init_range": 0.02,