han_config.py

import os
import data

file_path = os.path.dirname(os.path.realpath(__file__))

class HierarchicalAttentionConfig(object):
    def __init__(self):
        self.learning_rate = 0.001
        self.num_sentence = 10 # num sentence in documents
        self.sentence_length = 100 # sequence length of each sentence
        self.hidden_size = 300
        self.data_set_path = ''
        self.gru_output_keep_prob = 0.5
        self.class_num = 2
        self.batch_size = 8
        self.epoch = 10  # train epoch
        self.train_rate = 0.8 # this is train data set rate of total data set
        self.num_train = 5 # how many number display train accuracy
        self.require_improved = 100

        if os.path.exists(file_path + '/model'):
            print('model path exist....')
        else:
            os.mkdir(file_path + '/model/')
        self.model_path = file_path + '/model/HAN.ckpt'

        if os.path.exists(file_path + '/log'):
            print('log path exist....')
        else:
            os.mkdir(file_path + '/log')
        self.log_path = file_path + '/log'