Update README.md

README.md

@@ -14,8 +14,13 @@ language:
 
 This model is capable of determining the definition of the homonym "ბარი" located at the position marked by the [MASK] token.
 
-It is a simple Transformer model trained on a hand classified dataset comprising 6000 hand-classified sentences.
+It is a simple Transformer model fine-tuned on a hand classified dataset comprising 4800 hand-classified sentences.
 
+It shows 95% accuracy on a test set comprising 1200 hand-classified sentences. 
+
+The original 6000 sentences were split into 80% training data and 20% testing data. <a href="https://huggingface.co/datasets/davmel/ka_homonym_disambiguation">link to dataset</a>
+
+<h1>methodology:</h1>
 I've masked the homonyms from the sentences and replaced them with their synonyms according to the definitions used. For example, I replaced ”ბარი” with ”დაბლობი” (lowland) where the homonym referred to the field.
 
 The model predicts "თო" when it interprets the homonym as "Shovel," "დაბ" when it interprets it as "lowland," and "კაფე" when it interprets it as "Cafe."