Use CNN's to predict user behavior on mobile UI's.
We assume user interactions are broken down into 7 classes:
- Do nothing
- Tap top left quadrant
- Tap top right quadrant
- Tap bottom left quadrant
- Tap bottom right quadrant
- Swipe left/right
- Swipe up/down
Our training set consists of 10000 UI's for each class. Our testing set consists of 500 UI's for each class.
Usage:
- Download the RICO interaction traces dataset (http://interactionmining.org/rico) and put it in this directory.
- Run 'python3 generate_labels.py' to generate the screenshot-to-gesture labels and preprocess images. This can take a while.
- Run 'python3 train.py', adjusting the step size as the model converges. We recommend starting with a step size of 0.001.
- Run 'python3 test.py'.
- To visualize ROC curves, run 'python3 plotting/plot_ROC_curves.py'.
- To visualize the confusion matrix, run 'python3 plotting/plot_confusion_matrix.py'