Skip to content

nissymori/PUORL

Repository files navigation

Offline Reinforcement Learning with Domain-Unlabeled Data

arXiv

Official implementation of Offline Reinforcement Learning with Domain-Unlabeled Data accepted at RLC 2025. The offline RL implementations are borrowed from JAX-CORL.

How to install

git clone https://github.com/nissymori/PUORL.gitn
pip install -r requirements.txt

How to reproduce

  1. Download data
python download.py
  1. Remove prefix
./remove_prefix.sh
  1. Train classifier at sh/classifier
./pu.sh
  1. Train agent at sh/offline/...
./body_mass.sh

Wandb results for reported result

Citeation

@article{nishimori2025offline,
    title={Offline Reinforcement Learning with Domain-Unlabeled Data},
    author={Nishimori, Soichiro and Cai, Xin-Qiang and Ackermann, Johannes and Sugiyama, Masashi},
    journal={Reinforcement Learning Journal},
    volume={6},
    pages={1773--1793},
    year={2025}
}

About

[RLC2025] Official code for "Offline Reinforcement Learning with Domain-Unlabeled Data"

Resources

Stars

5 stars

Watchers

1 watching

Forks

Releases

No releases published

Packages

 
 
 

Contributors