https://github.com/tensorflow/models/tree/archive/research/deep_contextual_bandits