Update README.md

README.md

@@ -42,7 +42,7 @@ datasets:
 ---
 # Model description
 This `bert-causation-rating-dr2` model is a fine-tuned [biobert-base-cased-v1.2](https://huggingface.co/dmis-lab/biobert-base-cased-v1.2) model on a small set of manually annotated texts with causation labels. This model is tasked with classifying a sentence into different levels of strength of causation expressed in this sentence.
-
+Before tuning on this dataset, the `biobert-base-cased-v1.2` model is fine-tuned on a dataset containing causation labels from a published paper. This model starts from pre-trained [`kelingwang/bert-causation-rating-pubmed`](https://huggingface.co/kelingwang/bert-causation-rating-pubmed). For more information please view the link and my [GitHub page](https://github.com/Keling-Wang/causation_rating).
 The sentences in the dataset were rated independently by two researchers. This `dr2` version is tuned on the set of sentences with labels rated by Rater 2 and 3. 
 
 # Intended use and limitations
@@ -86,6 +86,7 @@ This performance is achieved with the following hyperparameters:
 # Training settings
 
 The following training configurations apply:
+ * Pre-trained model: `kelingwang/bert-causation-rating-pubmed`
  * `seed`: 114514
  * `batch_size`: 128
  * `epoch`: 8