import json


class Config(object):
    def __init__(self, args):
        self.args = args
        self.lr = args.lr
        self.dataset = args.dataset
        self.batch_size = args.batch_size
        self.max_epoch = args.max_epoch
        self.max_len = args.max_len
        self.bert_name = args.bert_name
        self.bert_dim = args.bert_dim

        self.train_path = 'data/' + self.dataset + '/train.json'
        # self.dev_path = 'data/' + self.dataset + '/dev.json'
        self.test_path = 'data/' + self.dataset + '/test.json'
        self.rel_path = 'data/' + self.dataset + '/rel.json'
        self.num_relations = len(json.load(open(self.rel_path, 'r')))

        self.save_weights_dir = 'saved_weights/'
        self.save_logs_dir = 'saved_logs/'
        self.result_dir = 'results/'

        self.period = 50
        self.weights_save_name = self.dataset + '_model.pkl'
        self.log_save_name = self.dataset + '_model.log'
        self.result_save_name = self.dataset + '_result.json'

b0fecdde0ad16d510d55fd42737fb04bec595f00
<mask token>

<mask token>


class Config(object):
    <mask token>

<mask token>


class Config(object):

    def __init__(self, args):
        self.args = args
        self.lr = args.lr
        self.dataset = args.dataset
        self.batch_size = args.batch_size
        self.max_epoch = args.max_epoch
        self.max_len = args.max_len
        self.bert_name = args.bert_name
        self.bert_dim = args.bert_dim
        self.train_path = 'data/' + self.dataset + '/train.json'
        self.test_path = 'data/' + self.dataset + '/test.json'
        self.rel_path = 'data/' + self.dataset + '/rel.json'
        self.num_relations = len(json.load(open(self.rel_path, 'r')))
        self.save_weights_dir = 'saved_weights/'
        self.save_logs_dir = 'saved_logs/'
        self.result_dir = 'results/'
        self.period = 50
        self.weights_save_name = self.dataset + '_model.pkl'
        self.log_save_name = self.dataset + '_model.log'
        self.result_save_name = self.dataset + '_result.json'

import json


class Config(object):

    def __init__(self, args):
        self.args = args
        self.lr = args.lr
        self.dataset = args.dataset
        self.batch_size = args.batch_size
        self.max_epoch = args.max_epoch
        self.max_len = args.max_len
        self.bert_name = args.bert_name
        self.bert_dim = args.bert_dim
        self.train_path = 'data/' + self.dataset + '/train.json'
        self.test_path = 'data/' + self.dataset + '/test.json'
        self.rel_path = 'data/' + self.dataset + '/rel.json'
        self.num_relations = len(json.load(open(self.rel_path, 'r')))
        self.save_weights_dir = 'saved_weights/'
        self.save_logs_dir = 'saved_logs/'
        self.result_dir = 'results/'
        self.period = 50
        self.weights_save_name = self.dataset + '_model.pkl'
        self.log_save_name = self.dataset + '_model.log'
        self.result_save_name = self.dataset + '_result.json'

import json


class Config(object):
    def __init__(self, args):
        self.args = args
        self.lr = args.lr
        self.dataset = args.dataset
        self.batch_size = args.batch_size
        self.max_epoch = args.max_epoch
        self.max_len = args.max_len
        self.bert_name = args.bert_name
        self.bert_dim = args.bert_dim

        self.train_path = 'data/' + self.dataset + '/train.json'
        # self.dev_path = 'data/' + self.dataset + '/dev.json'
        self.test_path = 'data/' + self.dataset + '/test.json'
        self.rel_path = 'data/' + self.dataset + '/rel.json'
        self.num_relations = len(json.load(open(self.rel_path, 'r')))

        self.save_weights_dir = 'saved_weights/'
        self.save_logs_dir = 'saved_logs/'
        self.result_dir = 'results/'

        self.period = 50
        self.weights_save_name = self.dataset + '_model.pkl'
        self.log_save_name = self.dataset + '_model.log'
        self.result_save_name = self.dataset + '_result.json'
