Export named signatures from a tf.contrib.learn estimator

export.py

# XXX: Only work with tf 11

from tensorflow.contrib.learn.python.learn.utils import export
from tensorflow.contrib.session_bundle import exporter

def my_input_fn():
    # Return features tensors, the name is empty because when we build the model
    # we feed in empty column name
    return {
        "": tf.placeholder(tf.float32, shape=[None, 4])
    }, None

def my_signature_fn(examples,features,predictions):
    return None,{
        "inputs": exporter.generic_signature({"features": examples}),
        "outputs": exporter.generic_signature({"score": predictions})
    }

classifier.export(
    "/tmp/iris_exported_model",
    input_fn=my_input_fn,
    input_feature_key="",
    use_deprecated_input_fn=False,
    signature_fn=my_signature_fn
)