Update README.md

README.md

@@ -1,3 +1,17 @@
 ---
 license: mit
 ---
+
+Three classes sentiment analysis (positive, negative, neutral)
+
+Based on https://huggingface.co/siebert/sentiment-roberta-large-english
+
+Fine-tuned using:
+- annotated sentences from book reviews in English https://www.gti.uvigo.es/index.php/en/book-reviews-annotated-dataset-for-aspect-based-sentiment-analysis
+- annotated paragraphs from amateur writers' stories https://arxiv.org/abs/1910.11769
+
+Performance for books:
+![hartmann_ft_test_books.png](https://s3.amazonaws.com/moonup/production/uploads/1661430297130-611a6e0f289467cafea62d0e.png)
+
+Performance for reviews:
+![hartmann_ft_test_reviews.png](https://s3.amazonaws.com/moonup/production/uploads/1661430338990-611a6e0f289467cafea62d0e.png)