def translate(config_path, input_path, output_path): print('Loading config ... ', end='', flush=True) with open(config_path, 'r') as f: config = json.load(f) print('done !') dic = Dic() dic.read_dict(config['dic'], config['word'], True) translater = Translater(dic, config['model']) if input_path == '': translater.shell() else: translater.translate_file(input_path, output_path) return
def train(config_path): jieba.initialize() # init jieba print('Loading training config ... ', end='', flush=True) with open(config_path, 'r') as f: config = json.load(f) print('done !') dic = Dic() dic.read_dict(config['dic'], config['word']) trainer = Trainer(dic) # trainer.feed(config['word'], True) for data in config['data']: trainer.feed(data) trainer.build() trainer.write_into_file(config['model'])