From 9b65b87e626e80ddf7481828a28f93ab6ea3ec39 Mon Sep 17 00:00:00 2001
From: Kolby Nottingham <kolbytn@gmail.com>
Date: Tue, 6 Feb 2024 12:20:12 -0800
Subject: [PATCH] links and cite in readme

---
 README.md | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 4fb9651..bda989b 100644
--- a/README.md
+++ b/README.md
@@ -1,5 +1,13 @@
 # Skill Set Optimization
 
+<a href="https://allenai.github.io/sso/">
+    <img src="https://img.shields.io/badge/Project Page-red">
+  </a>
+  <a href="https://arxiv.org/pdf/2402.03244.pdf">
+    <img src="https://img.shields.io/badge/Paper-blue">
+  </a> 
+</p>
+
 ## Installation
 
 pip install python library requirements with:
@@ -79,4 +87,15 @@ Evaluate on a ScienceWorld task and variants using:
 --train_iters 3
 ```
 
-Note that `task` can be set to any valid ScienceWorld task, and variants can be selected randomly with `train_variant_count`/`test_variant_count` or by specifying specific variants with `train_variants`/`test_variants`.
\ No newline at end of file
+Note that `task` can be set to any valid ScienceWorld task, and variants can be selected randomly with `train_variant_count`/`test_variant_count` or by specifying specific variants with `train_variants`/`test_variants`.
+
+## Citation
+```bib
+@article{nottingham2024sso,
+  author    = "Nottingham, Kolby and Majumder, Bodhisattwa Prasad and Dalvi Mishra, Bhavana and Singh, Sameer and Clark, Peter and Fox, Roy",
+  title     = "Skill Set Optimization: Reinforcing Language Model Behavior via Transferable Skills",
+  journal   = "arXiv",
+  year      = "2024",
+  url       = "https://arxiv.org/abs/2402.03244"
+}
+```
\ No newline at end of file