当前位置: 首页>>代码示例>>Python>>正文


Python cnn_model.TextCNN方法代码示例

本文整理汇总了Python中cnn_model.TextCNN方法的典型用法代码示例。如果您正苦于以下问题:Python cnn_model.TextCNN方法的具体用法?Python cnn_model.TextCNN怎么用?Python cnn_model.TextCNN使用的例子?那么, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在cnn_model的用法示例。


在下文中一共展示了cnn_model.TextCNN方法的3个代码示例,这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞,您的评价将有助于系统推荐出更棒的Python代码示例。

示例1: load_model

# 需要导入模块: import cnn_model [as 别名]
# 或者: from cnn_model import TextCNN [as 别名]
def load_model(self):
    sess = tf.Session()
    print('Configuring CNN model...')
    config = TCNNConfig()
    cnn_model = TextCNN(config)

    saver = tf.train.Saver()
    params_file = tf.train.latest_checkpoint(self.model_dir)
    saver.restore(sess, params_file)

    categories, cat_to_id = read_category()
    vocab_dir = 'cnews/cnews.vocab.txt'
    words, word_to_id = read_vocab(vocab_dir)

    self.words = words
    self.word_to_id = word_to_id
    self.categories = categories
    self.cat_to_id = cat_to_id

    self.cnn_model = cnn_model
    self.sess = sess
    print(self.cnn_model)
    print(self.sess) 
开发者ID:ucloud,项目名称:uai-sdk,代码行数:25,代码来源:txt_cnn_rnn_inference.py

示例2: __init__

# 需要导入模块: import cnn_model [as 别名]
# 或者: from cnn_model import TextCNN [as 别名]
def __init__(self):
        self.config = TCNNConfig()
        self.categories, self.cat_to_id = read_category()
        self.words, self.word_to_id = read_vocab(vocab_dir)
        self.config.vocab_size = len(self.words)
        self.model = TextCNN(self.config)

        self.session = tf.Session()
        self.session.run(tf.global_variables_initializer())
        saver = tf.train.Saver()
        saver.restore(sess=self.session, save_path=save_path)  # 读取保存的模型 
开发者ID:a414351664,项目名称:Bert-TextClassification,代码行数:13,代码来源:predict.py

示例3: __init__

# 需要导入模块: import cnn_model [as 别名]
# 或者: from cnn_model import TextCNN [as 别名]
def __init__(self):
        self.config = TCNNConfig()
        self.categories, self.cat_to_id = read_category()
        self.word_to_id = read_vocab(vocab_dir)
        self.config.vocab_size = len(self.word_to_id)
        self.model = TextCNN(self.config)

        self.session = tf.Session()
        self.session.run(tf.global_variables_initializer())
        saver = tf.train.Saver()
        saver.restore(sess=self.session, save_path=save_path)  # 读取保存的模型 
开发者ID:sliderSun,项目名称:pynlp,代码行数:13,代码来源:predict_cnn.py


注:本文中的cnn_model.TextCNN方法示例由纯净天空整理自Github/MSDocs等开源代码及文档管理平台,相关代码片段筛选自各路编程大神贡献的开源项目,源码版权归原作者所有,传播和使用请参考对应项目的License;未经允许,请勿转载。