Skip to content

Commit

Permalink
links and cite in readme
Browse files Browse the repository at this point in the history
  • Loading branch information
kolbytn committed Feb 6, 2024
1 parent 06a1b1c commit 9b65b87
Showing 1 changed file with 20 additions and 1 deletion.
21 changes: 20 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
@@ -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:
Expand Down Expand Up @@ -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`.
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"
}
```

0 comments on commit 9b65b87

Please sign in to comment.