Update README.md

README.md

@@ -14,12 +14,27 @@ of labels that yielded better predictions (see notebook [here](https://www.kaggl
 
 * issue
 * feature request
-* question 
+* question
 
 
 ## Training data
 
-This model was trained on 5k titles (unlabelled, as per distillation requierements) from the train dataset of [Github Issues Prediction dataset](https://www.kaggle.com/datasets/anmolkumar/github-bugs-prediction)
+* 15k of Github issues titles ("unlabeled_titles_simple.txt")
+* Hypothesis used: "This request is a {}"
+* Teacher model used: valhalla/distilbart-mnli-12-1
+* Studend model used: distilbert-base-uncased
+
+## Results
+
+Agreement of student and teacher predictions: **94.82%**
+
+See [this notebook](https://www.kaggle.com/code/antoinemacia/zero-shot-classifier-for-bug-analysis/edit) for more info on feature engineering choice made
+
+
+## How to train using your own dataset
+* Download training dataset from https://www.kaggle.com/datasets/anmolkumar/github-bugs-prediction
+* Modify and run convert.py, updating the paths to convert to a CSV
+* Run distill.py with the csv file (see [here](https://github.com/huggingface/transformers/tree/main/examples/research_projects/zero-shot-distillation) for more info)
 
 ## Acknowledgements