yes, predict() is part of the Estimator interface:

https://www.tensorflow.org/api_docs/python/tf/estimator/Estimator

so it will work. Make sure to provide an export function signature to Experiment so that the saved model can do online predictions