def __init__(self, metric: str = 'char'):
self.target_list = list()
self.predict_list = list()
if metric == 'char':
self.metric = CharacterErrorRate(id2char, EOS_token)
elif metric == 'word':
self.metric = WordErrorRate(id2char, EOS_token)
else:
raise ValueError("Unsupported metric : {0}".format(metric))
def __init__(self, vocab, metric: str = 'char'):
self.target_list = list()
self.predict_list = list()
self.vocab = vocab
if metric == 'char':
self.metric = CharacterErrorRate(vocab)
elif metric == 'word':
self.metric = WordErrorRate(vocab)
else:
raise ValueError("Unsupported metric : {0}".format(metric))