diff --git a/ELMo/LAC_demo/network.py b/ELMo/LAC_demo/network.py index 96471fa12b609a8cd728260c3f77a2f0f4dc29bc..c982c2538b69e7cd7fcdf41485ee86330e552cd5 100755 --- a/ELMo/LAC_demo/network.py +++ b/ELMo/LAC_demo/network.py @@ -9,7 +9,6 @@ import paddle.fluid as fluid from paddle.fluid.initializer import NormalInitializer import paddle.fluid.layers as layers from bilm import elmo_encoder -import ipdb def lex_net(args, word_dict_len, label_dict_len): diff --git a/ERNIE/finetune/classifier.py b/ERNIE/finetune/classifier.py index 950c550944e4c3cb3fb2e3f6091b8a45aaeee08e..e7a79ac8df09d0d8266b26ff0f3876d65427afee 100644 --- a/ERNIE/finetune/classifier.py +++ b/ERNIE/finetune/classifier.py @@ -177,6 +177,8 @@ def evaluate(exe, test_program, test_pyreader, graph_vars, eval_phase): total_acc += np.sum(np_acc * np_num_seqs) total_num_seqs += np.sum(np_num_seqs) labels.extend(np_labels.reshape((-1)).tolist()) + if np_qids is None: + np_qids = np.array([]) qids.extend(np_qids.reshape(-1).tolist()) scores.extend(np_probs[:, 1].reshape(-1).tolist()) np_preds = np.argmax(np_probs, axis=1).astype(np.float32) diff --git a/ERNIE/reader/task_reader.py b/ERNIE/reader/task_reader.py index c00bdb51f575395c5f437ec8f9d815be2d60ad52..5dab909f147160a1ebfb7286e79a251c4047202e 100644 --- a/ERNIE/reader/task_reader.py +++ b/ERNIE/reader/task_reader.py @@ -238,7 +238,7 @@ class ClassifyReader(BaseReader): batch_labels = [record.label_id for record in batch_records] batch_labels = np.array(batch_labels).astype("int64").reshape([-1, 1]) - if batch_records[0].qid: + if batch_records[0].qid is not None: batch_qids = [record.qid for record in batch_records] batch_qids = np.array(batch_qids).astype("int64").reshape([-1, 1]) else: