diff --git a/README.md b/README.md index 4a7f37c..fb59cb3 100644 --- a/README.md +++ b/README.md @@ -15,7 +15,7 @@ -| Features | Model List | Dataset | Quick Start | Documentation | Citation |Acknowledgement | Star History | +| Features | Model List | Dataset | Quick Start | Benchmark |Documentation | Citation |Acknowledgement | Star History | ## News @@ -169,6 +169,19 @@ NOTE: The full documentation is available on the [website](https://ant-research.github.io/EasyTemporalPointProcess/). +## Benchmark [Back to Top] + + +In the [examples](https://github.com/ant-research/EasyTemporalPointProcess/tree/main/examples) folder, we provide a [script](https://github.com/ant-research/EasyTemporalPointProcess/blob/main/examples/benchmark_script.py) to benchmark the TPPs, with Taxi dataset as the input. + +To run the script, one should download the Taxi data following the above instructions. The [config](https://github.com/ant-research/EasyTemporalPointProcess/blob/main/examples/configs/experiment_config.yaml) file is readily setup up. Then run + + +```shell +cd examples +python benchmark_script.py +``` + ## License [Back to Top]