DOC: Make predict docs templated