This project is a code replication of "Cellular QoE Prediction for Video Service Based on Causal Structure Learning" as described in the published work. As the original paper did not release the used dataset, the replication has been tested on synthetically generated random data.
These instructions will get you a copy of the project up and running on your local machine for development and testing purposes.
Before you proceed, make sure you have installed the necessary packages and libraries. You should have Python>3.7 installed on your machine. The specific python libraries needed should be installed via pip:
pip install -r requirements.txt
To generate synthetic KPI/KQI data, run the following command:
python gen.py
This will generate the data in the ./data/
directory. You should be able to find the generated dataset at ./data/cellular/cellular.csv
.
You can modify the network parameters and training hyperparameters in the configs.py
file. Please consult this file and make changes as necessary for your specific application.
To start the training and testing process, run the following command:
python main.py
This will start the training process and test the model after training.
The best-performing checkpoint from all epochs on the validation set, as well as the test performance of the model, will be saved in best_model.pt
.
- Shaoyu Dou (窦绍瑜) - Ph.D student, Tongji University, China.
This project is licensed under the MIT License - see the LICENSE file for details
For any questions or concerns, feel free to reach out to us. You can also raise an issue via GitHub's issue tracking system. If you need direct contact, please email the project maintainer at [email protected].
- This project has been tested on synthetically generated data due to the unavailability of the original dataset.
- The following open-source projects have provided inspiration and code which this project builds upon: