tatk.policy.mdrg.multiwoz.utils package

Submodules

tatk.policy.mdrg.multiwoz.utils.dbPointer module

tatk.policy.mdrg.multiwoz.utils.dbPointer.oneHotVector(num, domain, vector)

Return number of available entities for particular domain.

tatk.policy.mdrg.multiwoz.utils.dbPointer.queryResult(domain, turn)

Returns the list of entities for a given domain based on the annotation of the belief state

tatk.policy.mdrg.multiwoz.utils.dbPointer.queryResultVenues(domain, turn, real_belief=False)
tatk.policy.mdrg.multiwoz.utils.dbPointer.table_schema(domain)

tatk.policy.mdrg.multiwoz.utils.dbquery module

tatk.policy.mdrg.multiwoz.utils.dbquery.auto_download()
tatk.policy.mdrg.multiwoz.utils.dbquery.query(domain, constraints, ignore_open=True)

Returns the list of entities for a given domain based on the annotation of the belief state

tatk.policy.mdrg.multiwoz.utils.delexicalize module

tatk.policy.mdrg.multiwoz.utils.delexicalize.delexicalise(utt, dictionary)
tatk.policy.mdrg.multiwoz.utils.delexicalize.delexicaliseDomain(utt, dictionary, domain)
tatk.policy.mdrg.multiwoz.utils.delexicalize.prepareSlotValuesIndependent()

tatk.policy.mdrg.multiwoz.utils.nlp module

class tatk.policy.mdrg.multiwoz.utils.nlp.BLEUScorer

Bases: object

__init__()

Initialize self. See help(type(self)) for accurate signature.

score(hypothesis, corpus, n=1)
class tatk.policy.mdrg.multiwoz.utils.nlp.GentScorer(detectfile)

Bases: object

__init__(detectfile)

Initialize self. See help(type(self)) for accurate signature.

scoreBLEU(parallel_corpus)
tatk.policy.mdrg.multiwoz.utils.nlp.insertSpace(token, text)
tatk.policy.mdrg.multiwoz.utils.nlp.normalize(text)
tatk.policy.mdrg.multiwoz.utils.nlp.sentence_bleu_4(hyp, refs, weights=[0.25, 0.25, 0.25, 0.25])

tatk.policy.mdrg.multiwoz.utils.util module

Utility functions

tatk.policy.mdrg.multiwoz.utils.util.asMinutes(s)
tatk.policy.mdrg.multiwoz.utils.util.loadDialogue(model, val_file, input_tensor, target_tensor, bs_tensor, db_tensor)
tatk.policy.mdrg.multiwoz.utils.util.load_config(basename)
tatk.policy.mdrg.multiwoz.utils.util.load_dict(filename)
tatk.policy.mdrg.multiwoz.utils.util.padSequence(tensor)
tatk.policy.mdrg.multiwoz.utils.util.str2bool(v)
tatk.policy.mdrg.multiwoz.utils.util.timeSince(since, percent)
tatk.policy.mdrg.multiwoz.utils.util.unicode_to_utf8(d)