Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion preprocessing/sports/SAR_data/soccer/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@ You can find detailed documentation on supported data providers [here](https://o
For information on supported preprocessing methods, visit [this documentation](https://openstarlab.readthedocs.io/en/latest/Pre_Processing/Sports/SAR_data/Data_Format/Soccer/index.html). The available preprocessing methods are:

- State Action Reward (SAR) Format
- SAR-to-RL Dataset Conversion (SAR2RL)

## Examples
Here are some examples of how to download and preprocess data:
Expand All @@ -29,10 +30,12 @@ Here are some examples of how to download and preprocess data:
## SAR-to-RL Dataset Conversion (DQN / QMIX)
This section describes a SAR-to-RL dataset conversion step that formats SAR outputs (`events.jsonl`) into tensors used by
DQN and QMIX training. This is a preprocessing/data-format step, not a training algorithm.
The conversion script is `soccer_sar_to_rl_dataset.py`.

This produces a single shared multi-agent dataset with:
- `observation`: `(B, T, N, O)` (N=10 attackers)
- `action`: `(B, T, N)` (discrete action ids; default vocab size 16 with `PAD=15`)
- `reward`, `done`, `mask`: `(B, T)`
- `onball_mask`: `(B, T, N)` (for masking unavailable actions)

### Run via `SAR_data(...).preprocess_data()`
You can run SAR2RL through the same entry point as SAR