示例#1
0
def load_model():

    global nn, sess, trump

    if nn is None and sess is None and trump is None:
        nn = LSTM(max_lr=0.01,
                  min_lr=0.0005,
                  seq_len=57,
                  vocab_size=2402,
                  n_embedding=1000,
                  n_out=2402,
                  n_cell=1000,
                  rand_seed=1234)
        trump = trump_data.TRUMP(
            trump_data.trump_tokenized_dataset_path_default,
            trump_data.trump_token_dict_path_default)
        tf_config = tf.ConfigProto()
        tf_config.gpu_options.allow_growth = True
        init = tf.global_variables_initializer()
        sess = tf.InteractiveSession(config=tf_config)
        sess.run(init)
        nn.build_arch()
        nn.restore_arch(sess)

    if nn is None or sess is None or trump is None:
        nn = None
        sess = None
        trump = None
        return False
    return True