nlp_base.py 文件源码

def __init__(self):
        self.model_name = self.__class__.__name__
        self.sess = tf.Session()
        # get arguments
        self.args = self.get_args()

        # log set
        logging.basicConfig(filename=self.args.log_file,
                            level=logging.DEBUG,
                            format='%(asctime)s %(message)s', datefmt='%y-%m-%d %H:%M')

        # set random seed
        np.random.seed(self.args.random_seed)
        tf.set_random_seed(self.args.random_seed)

        # save arguments
        save_args(args=self.args)