def __init__(self, params, name):
# We don't call the super constructor on purpose
#pylint: disable=W0231
"""Initializer"""
Configurable.__init__(self, params, tf.contrib.learn.ModeKeys.EVAL)
self._name = name
self._eos_token = self.params["eos_token"]
self._sos_token = self.params["sos_token"]
self._separator = self.params["separator"]
self._postproc_fn = None
if self.params["postproc_fn"]:
self._postproc_fn = locate(self.params["postproc_fn"])
if self._postproc_fn is None:
raise ValueError("postproc_fn not found: {}".format(
self.params["postproc_fn"]))
评论列表
文章目录