MCPcopy Create free account

hub / github.com/XuesongYang/end2end_dialog / functions

Functions92 in github.com/XuesongYang/end2end_dialog

↓ 30 callersFunctionwriteTxt
(data_lst, fname, prefix='', delimiter=None)
utils.py:230
↓ 22 callersFunctioncheckExistence
verify the existence of the data file
utils.py:19
↓ 12 callersMethodgetUserUtterMaxlen
(self)
DataSetCSVslotTagging.py:83
↓ 10 callersFunctiongetScore
(fname, prefix='')
analyze_log.py:12
↓ 9 callersFunctiongetActPred
make agent action prediction according to the threshold Inputs: act_probs: shape = (sample_nb, act_vocab_size), predicted action
utils.py:209
↓ 8 callersMethodpredict
return a list of lists, [[(w1, tag1), (w2, tag2), (w3, tag3)], [...], [...]]
BaselineModel.py:197
↓ 6 callersFunctioneval_intentPredict
Inputs: intent_probs: shape = (sample_nb, intent_vocab_size), predicted probs for intent prediction intent_trueLabel: shape =
utils.py:99
↓ 6 callersMethodtransform_data
(self, maxlen)
DataSetCSVslotTagging.py:91
↓ 6 callersFunctionwriteUtterActTxt
(utter_txt, act_txt, fname)
AgentActClassifyingModel.py:19
↓ 5 callersFunctioneval_actPred
make decision of label prediction given specific threshold
utils.py:56
↓ 5 callersFunctionget_windowedVec
padding (window_size -1) zero vectors in front of vecs, and chunking padded vecs iteratively using a sliding window. The chunking ra
utils.py:147
↓ 5 callersMethodload_model
(self, verbose=True)
BaselineModel.py:218
↓ 4 callersFunctioncalculate_FrameAccuracy
calculate frame-level accuracy = hit / sample_nb inputs: pred: shape = (sample_nb, dim_size), predicted ids matrix
utils.py:84
↓ 4 callersFunctioneval_slotTagging
Evaluation for slot tagging. Inputs: tag_probs: shape = (sample_nb, maxlen_userUtter, tag_vocab_size), predicted probs
utils.py:26
↓ 4 callersFunctiongetOrderedDict
get token2id, id2token, vocab_size for intent, act, tag, or word. need to reserve 1 for '<unk>', and reserve 0 for '<pad>'. note: th
DataSetCSV.py:18
↓ 3 callersFunctiongetBagOfWords
calculate BoW feature Input: an 2-darray user utterance with zero padding Output: 2-darray
BaselineModel.py:152
↓ 3 callersFunctiongetFrameScore
(tag_pred_fname, tag_target_fname, intent_pred_fname, intent_target_fname)
analyze_log.py:64
↓ 3 callersFunctionprint_params
(params_dct)
utils.py:142
↓ 3 callersFunctionstats
(fname)
csv_stats.py:14
↓ 3 callersFunctionto_categorical
transform into a 1hot matrix Input: y_seq: shape = (sample_nb, maxlen_userUtter), elements are token ids. nb_classes:
utils.py:66
↓ 3 callersMethodtransform_data
(self, maxlen)
DataSetCSVagentActPred.py:26
↓ 3 callersMethodtransform_data
(self, maxlen_userUtter)
DataSetCSVjoint.py:26
↓ 3 callersFunctionvectorizing_binaryVec
convert into binary vectors. Inputs: intents: shape = (sample_nb,) vocab_size: scalar, vocabulary size in
DataSetCSVslotTagging.py:47
↓ 3 callersFunctionwriteUtterTag
write user utter and user tags into a file. each line constrains with the format: w1 w2 w3\ttag1 tag2 tag3 Inputs: sents
BaselineModel.py:43
↓ 3 callersFunctionwriteUtterTagIntentTxt
(utter_txt, tag_txt, intent_txt, target_fname)
SlotTaggingModel_multitask.py:20
↓ 2 callersMethod_transform_data
convert textual utter and user tags into a list of lists that contain lists of (w, t) pairs
BaselineModel.py:186
↓ 2 callersFunctiongetFname
(fname, prefix='')
analyze_log.py:94
↓ 2 callersFunctiongetNLUframeAccuracy
(tag_probs, tag_mask, tag_trueLabel, intent_probs, intent_trueLabel, threshold)
utils.py:189
↓ 2 callersFunctiongetNLUpred
(tag_probs, tag_mask, id2tag, intent_probs, threshold, id2intent)
utils.py:169
↓ 2 callersFunctiongetTagPred
(tag_probs, tag_mask, id2tag)
utils.py:200
↓ 2 callersMethodload_model
(self)
AgentActClassifyingModel.py:161
↓ 2 callersFunctionload_model_NLU
(model_weights, test_data)
PipelineLstmModel.py:20
↓ 2 callersFunctionpipelinePrediction
(test_data, tag_model_weights, intent_model_weights, act_model_weights, result_folder, tuneTh=True, threshold=
PipelineLstmModel.py:80
↓ 2 callersMethodpredict
(self)
SlotTaggingModel_multitask.py:195
↓ 2 callersMethodpredict
(self)
JointModel_multitask_jointraining.py:236
↓ 2 callersMethodpredict
(self)
AgentActClassifyingModel.py:136
↓ 2 callersMethodtrain
(self, verbose=True)
BaselineModel.py:241
↓ 2 callersFunctionvectorizing_zeropad
encode utterance or slot tags into id sequence. 0s for padding, 1s for unk. return a matrix with shape=(sample_nb, maxlen) e.
DataSetCSVslotTagging.py:15
↓ 1 callersMethod_build
(self)
SlotTaggingModel_multitask.py:63
↓ 1 callersMethod_build
(self)
JointModel_multitask_jointraining.py:51
↓ 1 callersMethod_build
(self)
AgentActClassifyingModel.py:55
↓ 1 callersMethod_get_params
[Q] how to deal with the case: the tag, intent, act in test or dev set do not exist in train set, but they are predefined
DataSetCSV.py:100
↓ 1 callersMethod_load_data
(self)
DataSetCSV.py:72
↓ 1 callersMethod_plot_graph
(self)
SlotTaggingModel_multitask.py:187
↓ 1 callersMethod_plot_graph
(self)
JointModel_multitask_jointraining.py:303
↓ 1 callersMethod_plot_graph
(self)
AgentActClassifyingModel.py:128
↓ 1 callersFunctionbaselineScore
(fname)
analyze_log.py:23
↓ 1 callersFunctionbilstmOracleScore
(fname)
analyze_log.py:49
↓ 1 callersFunctioneval_tagPredBaseline
calculate performance score Input: y_true: true tags, [[(u'w1', tag1), (u'w2', tag2), (u'w3', tag3)], [...], [...]] y
BaselineModel.py:61
↓ 1 callersFunctiongetBaselineFnames
(baseline_fname)
analyze_log.py:140
↓ 1 callersFunctiongetPipelineFnames
(pipeline_fname)
analyze_log.py:111
↓ 1 callersFunctiongetTable
(train, dev, test)
csv_stats.py:40
↓ 1 callersFunctiongetTagBinaryVector
userTags_pred: [[(w1, tag1), (w2, tag2)],[]]
BaselineModel.py:33
↓ 1 callersFunctiongetUtterList
(sents)
BaselineModel.py:25
↓ 1 callersFunctionjointModelScore
(fname_tag, fname_intent, fname_act)
analyze_log.py:30
↓ 1 callersMethodload_model
(self)
SlotTaggingModel_multitask.py:252
↓ 1 callersMethodload_model
(self)
JointModel_multitask_jointraining.py:311
↓ 1 callersFunctionload_model_Policy
(model_weights)
PipelineLstmModel.py:34
↓ 1 callersFunctionmergeCSV
(csv_dct_lst)
iob2csv.py:60
↓ 1 callersFunctionmergeGuide
(dct_lst)
iob2csv.py:22
↓ 1 callersFunctionnluFrameScore
(baseline_fname, pipeline_fname, joint_tag_fname, joint_intent_fname)
analyze_log.py:79
↓ 1 callersFunctionpipelineBilstmScore
(fname)
analyze_log.py:54
↓ 1 callersFunctionprepare_agentActPredict
(csv_train, csv_test, csv_dev, npz_fname)
prepare_data.py:33
↓ 1 callersFunctionprepare_joint
(csv_train, csv_test, csv_dev, npz_fname)
prepare_data.py:52
↓ 1 callersFunctionprepare_slotTagging
(csv_train, csv_test, csv_dev, npz_fname)
prepare_data.py:12
↓ 1 callersFunctionreadIOB
(iob_file, utter_search)
iob2csv.py:150
↓ 1 callersFunctionreadIntentPredTxt
(intent_pred_txt, userIntent2id, sample_nb, userIntent_vocab_size)
PipelineLstmModel.py:63
↓ 1 callersFunctionreadTagPredTxt
(tag_pred_txt, userTag2id, sample_nb, userTag_vocab_size)
PipelineLstmModel.py:48
↓ 1 callersFunctionslotTaggingScore
(fname_tag, fname_intent)
analyze_log.py:40
↓ 1 callersFunctiontableEnd2End
(baseline, pipeline, jointModel, bilstmOracle)
analyze_log.py:169
↓ 1 callersFunctiontableNLU
(baseline, pipeline, jointModel, frame)
analyze_log.py:181
↓ 1 callersMethodtrain
(self, verbose=True)
BaselineModel.py:172
↓ 1 callersMethodtrain
(self)
SlotTaggingModel_multitask.py:105
↓ 1 callersMethodtrain
(self)
JointModel_multitask_jointraining.py:112
↓ 1 callersMethodtrain
(self)
AgentActClassifyingModel.py:77
↓ 1 callersFunctiontrainActModel
(train_data, dev_data, model_folder)
BaselineModel.py:135
↓ 1 callersFunctiontrainIntentModel
train intent prediction model using human annotated data Input: bag-of-words of user utterances, Output: indicator matrix of agent actions
BaselineModel.py:113
↓ 1 callersFunctiontrainSlotTaggingModel
train slot tagging model using human annotated data Input: userUtter Output: target userTags
BaselineModel.py:102
↓ 1 callersFunctiontransformLabelJson
construct a search dictionary: {sudialogID_utterIndex: normalized_utterance}
iob2csv.py:180
↓ 1 callersFunctionwriteCSV
(dct_lst, fname, fieldnames=None)
iob2csv.py:141
Method__init__
(self, **argparams)
BaselineModel.py:165
Method__init__
(self, **argparams)
BaselineModel.py:229
Method__init__
(self, csv_file, train_data=None, flag='train')
DataSetCSVslotTagging.py:79
Method__init__
(self, **argparams)
SlotTaggingModel_multitask.py:33
Method__init__
(self, **argparams)
JointModel_multitask_jointraining.py:19
Method__init__
(self, csv_file, window_size=5, train_data=None, flag='train')
DataSetCSVagentActPred.py:17
Method__init__
(self, csv_file, window_size=5, train_data=None, flag='train')
DataSetCSVjoint.py:17
Method__init__
(self, **argparams)
AgentActClassifyingModel.py:28
Method__init__
(self, csv_file, train_data=None, flag='train')
DataSetCSV.py:42
Methodload_model
(self, model_fname)
BaselineModel.py:290
FunctionplotLossConverge
Plot learning curves of losses, and save it as .png file. Input: loss_dct: {'train': [], 'val': [], 'train_tagging':[], 'tr
utils.py:120
Methodpredict
(self, X, y_vecBin, X_utter_txt, y_txt)
BaselineModel.py:270