exported_model.py 文件源码

def __init__(self, export_dir, model_name, use_spell_check=False, use_visual_features=False):
        '''
        Args:
            export_dir(type = str): Path to directory where trained model 
                                    has been exported (with trailing slash).
            model_name(type = str): Name of the model exported.
        '''
        self.export_dir = export_dir
        self.session = tf.Session()
        self.name = model_name
        self.use_spell_check = use_spell_check

        self.use_visual_features = use_visual_features