def main(conf_dict, use_cuda=False):
"""Train main function"""
data_generator = spo_data_reader.DataReader(
wordemb_dict_path=conf_dict['word_idx_path'],
postag_dict_path=conf_dict['postag_dict_path'],
label_dict_path=conf_dict['so_label_dict_path'],
p_eng_dict_path=conf_dict['label_dict_path'],
train_data_list_path=conf_dict['spo_train_data_path'],
test_data_list_path=conf_dict['spo_test_data_path'])
data_generator.load_train_data()
train(conf_dict, data_generator, use_cuda=use_cuda)
def main(conf_dict, use_cuda=False):
"""Train main function"""
if use_cuda and not fluid.core.is_compiled_with_cuda():
return
data_generator = spo_data_reader.DataReader(
wordemb_dict_path=conf_dict['word_idx_path'],
postag_dict_path=conf_dict['postag_dict_path'],
label_dict_path=conf_dict['so_label_dict_path'],
p_eng_dict_path=conf_dict['label_dict_path'],
train_data_list_path=conf_dict['spo_train_data_path'],
test_data_list_path=conf_dict['spo_test_data_path'])
train(conf_dict, data_generator, use_cuda=use_cuda)
def main(conf_dict, model_path, predict_data_path, predict_result_path, \
use_cuda=False):
"""Predict main function"""
data_generator = spo_data_reader.DataReader(
wordemb_dict_path=conf_dict['word_idx_path'],
postag_dict_path=conf_dict['postag_dict_path'],
label_dict_path=conf_dict['so_label_dict_path'],
p_eng_dict_path=conf_dict['label_dict_path'],
train_data_list_path=conf_dict['spo_train_data_path'],
test_data_list_path=conf_dict['spo_test_data_path'])
data_generator.load_test_data()
predict_infer(conf_dict, data_generator, predict_data_path, \
predict_result_path, model_path)