Tensorflow TFGAN GANEstimator出口模型

2018-5-09:目前GANEstimator没有创建所说的export_output词典; 它可以通过tensorflow/contrib/gan/python/estimator/python/head_impl.py162行后查看来验证.

您可以通过此GitHub Pull Request跟踪此问题的状态.

在model_fn需要返回一个EstimatorSpec与export_outputs字典.有点像:

if mode == Modes.PREDICT:
    predictions = {
        'classes': tf.gather(label_values, predicted_indices),
        'scores': tf.reduce_max(probabilities, axis=1)
    }
    export_outputs = {
        'prediction': tf.estimator.export.PredictOutput(predictions)
    }
    return tf.estimator.EstimatorSpec(
        mode, predictions=predictions, export_outputs=export_outputs)