fix bugs in metric

thu-coai · Jan 30, 2019 · faf1207 · faf1207
1 parent e97fee6
commit faf1207
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 19 deletions.
diff --git a/contk/metric/metric.py b/contk/metric/metric.py
@@ -315,13 +315,13 @@ class MultiTurnDialogRecorder(MetricBase):
 	Arguments:
 		dataloader (DataLoader): A dataloader for translating index to sentences.
 		context_key (str): Dialog context are passed to :func:`forward` by ``data[context_key]``.
-			Default: ``post``.
+			Default: ``context``.
 		reference_key (str): Dialog reference are passed to :func:`forward` by ``data[reference_key]``.
-			Default: ``resp``.
+			Default: ``reference``.
 		gen_key (str): Sentences generated by model are passed to :func:`forward` by
 			``data[gen_key]``. Default: ``gen``.
 	'''
-	def __init__(self, dataloader, context_key="content", reference_key="reference", gen_key="gen"):
+	def __init__(self, dataloader, context_key="context", reference_key="reference", gen_key="gen"):
 		super().__init__()
 		self.dataloader = dataloader
 		self.context_key = context_key

diff --git a/tests/metric/tes.log b/tests/metric/tes.log
diff --git a/tests/dataloader/test_metric.py → tests/metric/test_metric.py b/tests/dataloader/test_metric.py → tests/metric/test_metric.py
@@ -517,7 +517,7 @@ def test_close(self, argument, shape, type, batch_len, gen_len):
 		# 'equal' or 'unequal'
 		# 0, 1
 		dataloader = FakeDataLoader()
-		context_key, reference_key, gen_key = ('post', 'resp', 'gen') \
+		context_key, reference_key, gen_key = ('context', 'reference', 'gen') \
 			if argument == 'default' else ('ck', 'rk', 'gk')
 		data = dataloader.get_data(context_key=context_key, reference_key=reference_key, gen_key=gen_key, \
 								   multi_turn=True, to_list=(type == 'list'), pad=(shape == 'pad'), \