MCPcopy Create free account

hub / github.com/dasiki/Dialog-System-with-Task-Retrieval-and-Seq2seq / functions

Functions107 in github.com/dasiki/Dialog-System-with-Task-Retrieval-and-Seq2seq

↓ 215 callersMethodsearch
(cls, msg_tokens, mode="qa", filter_pattern=None)
search_dialog/search_core.py:30
↓ 7 callersFunctionlog_print
(text, level="DEBUG")
utils/tools.py:7
↓ 6 callersFunctionfile_path
(dirname, file_name)
seq2seq_dialog/config.py:8
↓ 6 callersFunctionlog_print
(text)
seq2seq_dialog/train.py:39
↓ 4 callersMethod_create_rnn_cell
(self)
seq2seq_dialog/model.py:32
↓ 4 callersFunction_load_words
(file_name)
utils/nlp_util.py:17
↓ 4 callersFunctionch_count
(text)
utils/tools.py:14
↓ 4 callersFunctionfile_path
(dirname, file_name)
search_dialog/config.py:7
↓ 4 callersFunctionloadDataset
(train_path=None, vocab_size=5000)
seq2seq_dialog/data_helpers.py:29
↓ 4 callersMethodtokenize
Tokenize text
utils/nlp_util.py:98
↓ 3 callersFunctionbleu
(answerFilePath, standardAnswerFilePath)
scripts/bleu.py:9
↓ 3 callersFunctioncreateBatch
(samples)
seq2seq_dialog/data_helpers.py:61
↓ 3 callersFunctionget_file_path
(file_name)
utils/global_names.py:30
↓ 3 callersFunctionget_infer_model
(dialog_mode)
seq2seq_dialog/infer.py:17
↓ 3 callersFunctionpredict
(model, context, ret_size=5)
seq2seq_dialog/infer.py:40
↓ 2 callersFunction_process_gen_msg
(text)
seq2seq_dialog/evaluate.py:13
↓ 2 callersFunctiongeneral_handle
(msg, dialog_status)
task_dialog/general_task.py:13
↓ 2 callersFunctionmulti_dialog
(model, questions)
seq2seq_dialog/evaluate.py:24
↓ 2 callersFunctionpredict_sent_emb
(model, context)
seq2seq_dialog/infer.py:47
↓ 2 callersFunctionsentence2enco
(sentence, word2id)
seq2seq_dialog/data_helpers.py:97
↓ 2 callersFunctiontext2id
(text, word2id)
seq2seq_dialog/data_helpers.py:56
↓ 2 callersMethodtrain
(self, batch)
seq2seq_dialog/model.py:178
↓ 1 callersFunction_input_streaming
(in_fpath, return_raw = False)
word2vec_model/train_word2vec.py:9
↓ 1 callersFunction_normalize_feature
(feature)
utils/rerank_util.py:16
↓ 1 callersFunction_predict_ids_to_seq
(predict_ids, id2word, beam_size)
seq2seq_dialog/infer.py:29
↓ 1 callersMethod_predict_via_seq2seq
(cls, msg_tokens)
dialog_manage.py:50
↓ 1 callersFunction_single_cell
Create an instance of a single RNN cell.
utils/model_helper.py:26
↓ 1 callersMethod_slots_update
(cls, msg, dialog_status)
task_dialog/task_core.py:63
↓ 1 callersMethod_text2vec
(self, text)
search_dialog/tfidf_model.py:50
↓ 1 callersMethod_train_model
(self, min_freq=1)
search_dialog/tfidf_model.py:37
↓ 1 callersFunctionbagging
(features, weights, normalize=True)
utils/rerank_util.py:24
↓ 1 callersFunctionbleu_score
(candidate, reference)
utils/bleu_util.py:11
↓ 1 callersMethodbuild_model
(self)
seq2seq_dialog/model.py:45
↓ 1 callersFunctiondump_word_embeddings
(word2id)
seq2seq_dialog/data_helpers.py:113
↓ 1 callersFunctiongenNextSamples
()
seq2seq_dialog/data_helpers.py:87
↓ 1 callersFunctiongetBatches
(data, batch_size)
seq2seq_dialog/data_helpers.py:83
↓ 1 callersMethodget_docs
(self, sim_items)
search_dialog/bm25_model.py:30
↓ 1 callersMethodget_tfidf
(cls, words)
utils/nlp_util.py:151
↓ 1 callersMethodinfer
(self, batch)
seq2seq_dialog/model.py:200
↓ 1 callersMethodinfer_sent_emb
(self, batch)
seq2seq_dialog/model.py:209
↓ 1 callersFunctioninvoiceReviseRule
dialogue stratege for invoice revise rule
task_dialog/invoice_task.py:158
↓ 1 callersFunctioninvoice_judge
(sentence)
task_dialog/invoice_task.py:40
↓ 1 callersMethodplace_recognize
(cls, text)
utils/nlp_util.py:70
↓ 1 callersMethodprocess_dialog
Dialog strategy: use sub-task to handle dialog firstly, if failed, use retrieval or generational func to handle it.
dialog_manage.py:58
↓ 1 callersMethodsearch
(cls, model, sent, size=10)
search_dialog/sent_emb_search.py:28
↓ 1 callersMethodsimilarity
(self, query, size=10)
utils/bm25_util.py:14
↓ 1 callersMethodsimilarity
(self, query, docs, size=10)
utils/wmd_util.py:17
↓ 1 callersMethodsimilarity
(self, query, size=10)
search_dialog/bm25_model.py:26
↓ 1 callersFunctionsingle_dialog
(model, context)
seq2seq_dialog/evaluate.py:18
↓ 1 callersFunctionstart_dialog
()
dialog_manage.py:95
↓ 1 callersMethodtask_handle
(cls, msg, dialog_status)
task_dialog/task_core.py:76
↓ 1 callersFunctiontrain_model
()
seq2seq_dialog/train.py:44
↓ 1 callersFunctiontrain_word2vec_model
(corpus_fpath, wv_fpath = None)
word2vec_model/train_word2vec.py:16
Method__init__
(self)
dialog_manage.py:15
Method__init__
(self)
seq2seq_dialog/data_helpers.py:22
Method__init__
(self, sess, mode, params, word_to_idx)
seq2seq_dialog/model.py:11
Method__init__
(self, corpus)
utils/bm25_util.py:9
Method__init__
(self, word2vec_path)
utils/wmd_util.py:10
Method__init__
(self, corpus_file, word2id)
search_dialog/bm25_model.py:15
Method__init__
(self, corpus_file, scale, word2id)
search_dialog/tfidf_model.py:19
Functionbleu_similarity
(query, docs)
utils/bleu_util.py:19
Functioncorpus2enco
(corpus, word2id)
seq2seq_dialog/data_helpers.py:107
Functioncreate_rnn_cell
Create multi-layer RNN cell.
utils/model_helper.py:8
Functiondelivery
(sentence,dialog_status)
task_dialog/delivery_task.py:13
Methodeval
(self, batch)
seq2seq_dialog/model.py:189
Functionevaluate
(model, dialog_mode)
seq2seq_dialog/evaluate.py:36
Functionfinish_handle
(sentence,dialog_status)
task_dialog/finish_task.py:22
Methodget_docs
(self, sim_items)
search_dialog/tfidf_model.py:63
Methodget_keywords
(cls, text, size=3, way=None)
utils/nlp_util.py:165
Functionget_model_tensor
Load model
utils/model_helper.py:60
Functioninfo_supply_handle
(msg, dialog_status)
task_dialog/info_supply_task.py:20
Functionintent_update
(msg, dialog_status)
task_dialog/start_task.py:22
Functionintent_update
(msg,dialog_status)
task_dialog/order_related_task.py:6
Functionintent_update
(msg, dialog_status)
task_dialog/order_modify.py:11
Functionintent_update
(msg,dialog_status)
task_dialog/delivery_task.py:3
Functionintent_update
(msg, dialog_status)
task_dialog/general_task.py:7
Functionintent_update
(msg, dialog_status)
task_dialog/orderid_ask_task.py:11
Functionintent_update
(msg,dialog_status)
task_dialog/finish_task.py:16
Functionintent_update
(msg, dialog_status)
task_dialog/info_supply_task.py:12
Functionintent_update
(msg, dialog_status)
task_dialog/short_query_task.py:15
Functionintent_update
(msg,dialog_status)
task_dialog/invoice_task.py:16
Functionintent_update
(msg,dialog_status)
task_dialog/sale_after_task.py:4
Functionintent_update
(msg,dialog_status)
task_dialog/sale_task.py:3
Functionintent_update
(msg,dialog_status)
task_dialog/refund_task.py:3
Functionintent_update
(msg, dialog_status)
task_dialog/price_protect_task.py:10
Functionintent_update
(msg,dialog_status)
task_dialog/query_task.py:3
Functionintent_update
(msg, dialog_status)
task_dialog/unbind_task.py:10
Functioninvoice_handle
(context, dialog_status)
task_dialog/invoice_task.py:23
Functionlength_weight
(corpus, orders, length_limit=6)
utils/tricks.py:33
Functionload_word_vector
(w2v_fpath)
word2vec_model/train_word2vec.py:41
Functionmodify_tokens
(tokens)
utils/tricks.py:9
Functionorder_consult
(msg, dialog_status)
task_dialog/order_consult_task.py:11
Functionorder_modify_handle
(msg, dialog_status)
task_dialog/order_modify.py:17
Functionorder_related
(sentence,dialog_status)
task_dialog/order_related_task.py:18
Functionorderid_ask_handle
(msg, dialog_status)
task_dialog/orderid_ask_task.py:19
Functionprice_protect_handle
(msg, dialog_status)
task_dialog/price_protect_task.py:17
Functionquery_judge
(sentence,dialog_status)
task_dialog/query_task.py:26
Functionrefund_response
(sentence,dialog_status)
task_dialog/refund_task.py:10
Functionrerank
(tfidf_feature, query, docs)
utils/rerank_util.py:41
Functionsale_after
(sentence,dialog_status)
task_dialog/sale_after_task.py:14
next →1–100 of 107, ranked by callers