model update

README.md

@@ -2,7 +2,7 @@
 datasets:
 - relbert/semeval2012_relational_similarity
 model-index:
-- name: relbert/relbert-roberta-large-nce-d-semeval2012
+- name: relbert/relbert-roberta-base-nce-semeval2012-0-400
   results:
   - task:
       name: Relation Mapping
@@ -186,11 +186,11 @@ model-index:
       value: 0.9040831832304929
 
 ---
-# relbert/relbert-roberta-large-nce-d-semeval2012
+# relbert/relbert-roberta-base-nce-semeval2012-0-400
 
 RelBERT based on [roberta-large](https://huggingface.co/roberta-large) fine-tuned on [relbert/semeval2012_relational_similarity](https://huggingface.co/datasets/relbert/semeval2012_relational_similarity) (see the [`relbert`](https://github.com/asahi417/relbert) for more detail of fine-tuning).
 This model achieves the following results on the relation understanding tasks:
-- Analogy Question ([dataset](https://huggingface.co/datasets/relbert/analogy_questions), [full result](https://huggingface.co/relbert/relbert-roberta-large-nce-d-semeval2012/raw/main/analogy.forward.json)):
+- Analogy Question ([dataset](https://huggingface.co/datasets/relbert/analogy_questions), [full result](https://huggingface.co/relbert/relbert-roberta-base-nce-semeval2012-0-400/raw/main/analogy.forward.json)):
     - Accuracy on SAT (full): 0.732620320855615 
     - Accuracy on SAT: 0.7359050445103857
     - Accuracy on BATS: 0.8093385214007782
@@ -200,13 +200,13 @@ This model achieves the following results on the relation understanding tasks:
     - Accuracy on ConceptNet Analogy: 0.4748322147651007
     - Accuracy on T-Rex Analogy: 0.644808743169399
     - Accuracy on NELL-ONE Analogy: 0.6583333333333333
-- Lexical Relation Classification ([dataset](https://huggingface.co/datasets/relbert/lexical_relation_classification), [full result](https://huggingface.co/relbert/relbert-roberta-large-nce-d-semeval2012/raw/main/classification.json)):
+- Lexical Relation Classification ([dataset](https://huggingface.co/datasets/relbert/lexical_relation_classification), [full result](https://huggingface.co/relbert/relbert-roberta-base-nce-semeval2012-0-400/raw/main/classification.json)):
     - Micro F1 score on BLESS: 0.9199939731806539
     - Micro F1 score on CogALexV: 0.8497652582159625
     - Micro F1 score on EVALution: 0.6836403033586133
     - Micro F1 score on K&H+N: 0.9563191208179731
     - Micro F1 score on ROOT09: 0.9041052961454089
-- Relation Mapping ([dataset](https://huggingface.co/datasets/relbert/relation_mapping), [full result](https://huggingface.co/relbert/relbert-roberta-large-nce-d-semeval2012/raw/main/relation_mapping.json)):
+- Relation Mapping ([dataset](https://huggingface.co/datasets/relbert/relation_mapping), [full result](https://huggingface.co/relbert/relbert-roberta-base-nce-semeval2012-0-400/raw/main/relation_mapping.json)):
     - Accuracy on Relation Mapping: 0.8049007936507937 
 
 
@@ -218,7 +218,7 @@ pip install relbert
 and activate model as below.
 ```python
 from relbert import RelBERT
-model = RelBERT("relbert/relbert-roberta-large-nce-d-semeval2012")
+model = RelBERT("relbert/relbert-roberta-base-nce-semeval2012-0-400")
 vector = model.get_embedding(['Tokyo', 'Japan'])  # shape of (n_dim, )
 ```
 
@@ -242,7 +242,7 @@ vector = model.get_embedding(['Tokyo', 'Japan'])  # shape of (n_dim, )
  - loss_function_config: {'temperature': 0.05, 'gradient_accumulation': 1, 'num_negative': 400, 'num_positive': 10}
  - augment_negative_by_positive: True
 
-See the full configuration at [config file](https://huggingface.co/relbert/relbert-roberta-large-nce-d-semeval2012/raw/main/finetuning_config.json).
+See the full configuration at [config file](https://huggingface.co/relbert/relbert-roberta-base-nce-semeval2012-0-400/raw/main/finetuning_config.json).
 
 ### Reference
 If you use any resource from RelBERT, please consider to cite our [paper](https://aclanthology.org/2021.emnlp-main.712/).