## Overview
This repository contains the source code for the paper

## Installation

* Extract the folder
* Run `cd feature4irl`
* Run `pip install -r requirements.txt`
* Optionally, setup [Weight and Biases](https://docs.wandb.ai/quickstart) for tracking the experiments.

## How to run the code
* Run `sh runs_exp.sh` for training the task respectively.
* Results of the experiments including the simulation videos, data will be then saved in `./checkpoints` folder.

## Configuration
Configuration files for each environment are located in ```feature4irl/cfg/```.

## To generate reward function:
* Run `cd feature4irl/util/`
* Run `python feature_create.py`


## Tutorials on how to use the reward construction are coming soon. 