Code for our WWW 19' paper
- Jianmo Ni, Larry Muhlstein, Julian McAuley, "Modeling heart rate and activity data for personalized fitness recommendation", in Proc. of the 2019 World Wide Web Conference (WWW'19), San Francisco, US, May. 2019.
You can find the data at here: https://sites.google.com/eng.ucsd.edu/fitrec-project/home. To run the code below, you need to first download the data files. Please cite our paper if you find the code and data helpful.
FitRec
This is the code for the workout profile prediction task. Data file you need: processed_endomondHR_proper.npy.
data_split.py- First run this to split the dataset into train/valid/test. Or you can directly download the files here endomondoHR_proper_temporal_dataset.pkl and endomondoHR_proper_metaData.pkl.heart_rate_aux.py- Run this file to predict the heart rate given the route and target time.speed_aux.py- Run this file to predict the speed given the route and target time.data_interpreter_Keras_aux.py- This is the dataloader file.
FitRec-Attn
This is the attention-based model for the short-term prediction task. Data file you need: processed_endomondoHR_proper_interpolate.npy.
data_split.py- Similarlry, first run this to split file. Or you can directly download the files here endomondoHR_proper_temporal_dataset.pkl.fitrect-attn.py- Run this to predict short-term prediction using our attention-based model.data_interpolate.py- This is the dataloader file.
Requirement
- Tensorflow>=1.4
- Keras=2.0
- PyTorch>=0.4