ageppert/world-model-transitions

---
language:
- en
license: mit
tags:
- world-model
- computer-use
- transition-prediction
---

# World Model Transition Dataset

Derived from the [AgentNet](https://huggingface.co/datasets/xlangai/AgentNet) Ubuntu 5K dataset
for training a world model (transition predictor) for computer-use agents.

## Format

Each example is a chat conversation:
- **System**: World model role description
- **User**: Current screen state observation + planned action
- **Assistant**: Predicted outcome (transition description)

## Source

Tier 1 extraction from AgentNet: steps with non-empty reflections,
correct actions, non-redundant, non-terminal.

See `preprocessing_stats.txt` for detailed filtering statistics.

## Citation

If you use this dataset, please cite the original OpenCUA paper:

```bibtex
@misc{wang2025opencua,
title={OpenCUA: Open Foundations for Computer-Use Agents},
author={Xinyuan Wang and others},
year={2025},
eprint={2508.09123},
archivePrefix={arXiv},
}
```