/fit-rec

Code for WWW 19' paper

Primary LanguagePython

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