MCPcopy Create free account

hub / github.com/DengBoCong/nlp-paper / functions

Functions229 in github.com/DengBoCong/nlp-paper

↓ 1 callersFunctionpositional_encoding_layer
(position, deep)
paper-code/tensorflow_src/models/gpt2.py:41
↓ 1 callersFunctionpreprocess_raw_lccc_data
用于处理LCCC数据集的方法,将LCCC数据集处理成问答对的形式 Args: raw_data: 原始数据路径 tokenized_data: 生成token数据保存路径 Returns:
paper-code/pytorch_src/common/pre_treat.py:67
↓ 1 callersFunctionpreprocess_raw_task_data
专门针对task标注数据的client和agent对话的token数据处理 :param raw_data: 原始对话数据路径 :param tokenized_data: 生成token数据保存路径 :return:
paper-code/tensorflow_src/models/task/common/pre_treat.py:117
↓ 1 callersFunctionprob_query_key
:param query: 查询 :param key: 键 :param sample_key: 采样个数 :param n_top: top数量 :return:
paper-code/tensorflow_src/models/informer.py:128
↓ 1 callersFunctionread_data
读取数据,将input和target进行分词后返回 :param path: Tokenizer文本路径 :param num_examples: 最大序列长度 :return: input_tensor, target_tensor, lang_tokenizer
paper-code/tensorflow_src/models/task/common/data_utils.py:56
↓ 1 callersFunctionread_tokenized_data
用于将分词文本读入内存,并整理成问答对,返回的是整理好的文本问答对以及权重 Args: path: 分词文本路径 start_sign: 开始标记 end_sign: 结束标记 num_examples: 读取的数据量
paper-code/pytorch_src/common/data_utils.py:36
↓ 1 callersFunctionrequest_slot_tracker
requestable插槽跟踪器,requestable插槽是用户询问系统的信息 用来获得时间t的状态的非分类插槽槽值分布, 比如: address=1 (地址被询问) phone=0 (用户不关心电话号码) 输入为状态跟踪器的输入'state_t'
paper-code/tensorflow_src/models/nbt.py:16
↓ 1 callersMethodrespond
(self, req)
paper-code/pytorch_src/model/chatter.py:102
↓ 1 callersFunctionscaled_dot_product_attention
计算注意力权重。 q, k, v 必须具有匹配的前置维度。 k, v 必须有匹配的倒数第二个维度,例如:seq_len_k = seq_len_v。 虽然 mask 根据其类型(填充或前瞻)有不同的形状, 但是 mask 必须能进行广播转换以便求和。 参数:
paper-code/tensorflow_src/tools/attention.py:58
↓ 1 callersFunctionscaled_dot_product_attention
点积注意力 :param queries: 请求的形状 == (..., seq_len_q, depth) :param keys: 主键的形状 == (..., seq_len_k, depth) :param values: 数值的形状 == (..., seq_l
paper-code/tensorflow_src/models/informer.py:103
↓ 1 callersFunctionscaled_dot_product_attention
计算注意力权重。 q, k, v 必须具有匹配的前置维度。 k, v 必须有匹配的倒数第二个维度,例如:seq_len_k = seq_len_v。 虽然 mask 根据其类型(填充或前瞻)有不同的形状, 但是 mask 必须能进行广播转换以便求和。 参数:
paper-code/tensorflow_src/models/InferSent.py:201
↓ 1 callersMethodsearch
(self, query: str, search_type: str)
search_kits.py:94
↓ 1 callersMethodsearch_multi
通过key批量查询知识对象 :params kvs: key和value的列表,使用lambda表达式进行累积操作
paper-code/tensorflow_src/models/task/common/kb.py:47
↓ 1 callersFunctionself_attention
(query, key, value, mask)
paper-code/tensorflow_src/models/gpt2.py:58
↓ 1 callersMethodset_paper_category_list
(self, categories_papers: Dict[str, List[Dict[str, Any]]])
search_kits.py:199
↓ 1 callersMethodshow
(self, title: str = "Paper Search Tool")
search_kits.py:264
↓ 1 callersFunctionstate_tracker
(units, vocab_size, embedding_dim, name="state_tracker")
paper-code/tensorflow_src/models/nbt.py:30
↓ 1 callersFunctiontask_encoder
task的encoder,使用双向LSTM对用户语句进行编码,输出序列和合并后的隐藏层
paper-code/tensorflow_src/models/nbt.py:62
↓ 1 callersFunctiontask_encoder
task的encoder,使用双向LSTM对用户语句进行编码,输出序列和合并后的隐藏层
paper-code/tensorflow_src/models/task/model/model.py:30
↓ 1 callersFunctiontokenize
分词方法,使用Keras API中的Tokenizer进行分词操作 :param input_lang: 输入 :param target_lang: 目标 :return: input_tensor, target_tensor, lang_tokenizer
paper-code/tensorflow_src/models/task/common/data_utils.py:68
↓ 1 callersMethodtrain
对模型进行训练
paper-code/pytorch_src/model/chatter.py:47
↓ 1 callersMethodtrain
(self, dict_fn, data_fn, start_sign, end_sign, max_train_data_size)
paper-code/tensorflow_src/models/task/task_chatter.py:37
↓ 1 callersFunctiontransformer_decoder_layer
(units, d_model, num_heads, dropout, name="transformer_decoder_layer")
paper-code/tensorflow_src/models/transformer.py:88
↓ 1 callersFunctiontransformer_encoder_layer
# Transformer的encoder层,使用函数式API :param units:单元大小 :param d_model:深度 :param num_heads:多头注意力的头部层数量 :param dropout:dropout的权重 :p
paper-code/tensorflow_src/models/transformer.py:54
↓ 1 callersFunctionunion
(key1: AnyStr, key2: AnyStr)
paper-code/data_enhancement.py:26
↓ 1 callersFunctionupdate_context
(context_in, value, scores, index)
paper-code/tensorflow_src/models/informer.py:162
FunctionGroupNorm
(x, gamma, beta, G, eps=1e-5)
paper-code/group_normalization.py:1
FunctionLogME
:param f: [N, F], feature matrix from pre-trained model :param y: target labels. For classification, y has shape [N] with element in
paper-code/pytorch_src/logME.py:45
FunctionLogME
(f: tf.Tensor, y: tf.Tensor, regression=False)
paper-code/tensorflow_src/logME.py:42
Method__call__
(self, attention_hidden_state, memory, attention_weights_cat)
paper-code/tensorflow_src/tools/attention.py:177
Method__call__
(self, step)
paper-code/tensorflow_src/models/gpt2.py:134
Method__call__
(self, step: Any)
paper-code/tensorflow_src/models/informer.py:12
Method__getitem__
(self, item)
paper-code/pytorch_src/common/data_utils.py:150
Method__init__
(self)
search_kits.py:29
Method__init__
Seq2Seq聊天器初始化,用于加载模型
paper-code/pytorch_src/seq2seq_chatter.py:21
Method__init__
(self, units: int)
paper-code/pytorch_src/model/seq2seq.py:32
Method__init__
(self, vocab_size: int, embedding_dim: int, enc_units: int, dec_units: int, dropout: float, a
paper-code/pytorch_src/model/seq2seq.py:58
Method__init__
聊天器初始化,用于加载模型
paper-code/pytorch_src/model/chatter.py:17
Method__init__
初始化BeamSearch的序列容器
paper-code/pytorch_src/common/utils.py:26
Method__init__
(self, input, target, diag_weight)
paper-code/pytorch_src/common/data_utils.py:145
Method__init__
(self, d_model, num_heads)
paper-code/tensorflow_src/tools/attention.py:97
Method__init__
(self, attention_n_filters, attention_kernel_size, attention_dim1)
paper-code/tensorflow_src/tools/attention.py:189
Method__init__
(self, units)
paper-code/tensorflow_src/models/seq2seq.py:5
Method__init__
(self, vocab_size, embedding_dim, dec_units, batch_sz)
paper-code/tensorflow_src/models/seq2seq.py:55
Method__init__
(self, d_model, warmup_steps=2000)
paper-code/tensorflow_src/models/gpt2.py:128
Method__init__
(self, d_model: Any, warmup_steps: Any = 4000)
paper-code/tensorflow_src/models/informer.py:6
Method__init__
(self, position, d_model)
paper-code/tensorflow_src/models/transformer.py:10
Method__init__
(self, d_model, num_heads)
paper-code/tensorflow_src/models/InferSent.py:237
Method__init__
(self, checkpoint_dir, beam_size)
paper-code/tensorflow_src/models/task/task_chatter.py:21
Method__init__
Transformer聊天器初始化,用于加载模型
paper-code/tensorflow_src/models/task/model/chatter.py:19
Method__init__
(self, info_slots, semi_dict, values, replaces)
paper-code/tensorflow_src/models/task/common/pre_treat.py:14
Method__init__
(self, dialogues, max_length, tokenizer, onto, onto_idx, max_train_data_size, kb_fonud_len=5,
paper-code/tensorflow_src/models/task/common/data_utils.py:256
Method__init__
(self, columns, primary)
paper-code/tensorflow_src/models/task/common/kb.py:23
Method__iter__
(self)
paper-code/tensorflow_src/models/task/common/data_utils.py:347
Method__len__
已存在BeamSearch的序列容器的大小
paper-code/pytorch_src/common/utils.py:34
Method__len__
(self)
paper-code/pytorch_src/common/data_utils.py:153
Method__len__
(self)
paper-code/tensorflow_src/models/task/common/data_utils.py:282
Method_create_predictions
(self, inputs, dec_input)
paper-code/pytorch_src/seq2seq_chatter.py:86
Method_create_predictions
(self, inputs, dec_input, t)
paper-code/tensorflow_src/models/task/task_chatter.py:34
Function_dollars_to_word
将美元转为单词 :param dollars_re: 美元匹配式 :return:
paper-code/tensorflow_src/tools/en_text_to_phoneme.py:173
Method_init_loss_accuracy
初始化损失
paper-code/pytorch_src/model/chatter.py:29
Method_init_loss_accuracy
(self)
paper-code/tensorflow_src/models/task/task_chatter.py:28
Function_number_to_word
将数字转为单词 :param number_re: 数字匹配式 :return:
paper-code/tensorflow_src/tools/en_text_to_phoneme.py:151
Function_parse_dataset_item
用于Dataset中的TFRecord序列化字符串恢复 :param example: 序列化字符串 :return: 恢复后的数据
paper-code/tensorflow_src/tools/preprocess_tfrecord.py:184
Method_train_step
(self, inp: torch.Tensor, tar: torch.Tensor, weight: torch.Tensor, teacher_forcing_ratio:
paper-code/pytorch_src/seq2seq_chatter.py:70
Method_train_step
(self, inp, tar, step_loss)
paper-code/tensorflow_src/models/task/task_chatter.py:31
Functionaccuracy
(real, pred)
paper-code/tensorflow_src/models/transformer.py:339
Functionbahdanau_attention
:param units: 全连接层单元数
paper-code/tensorflow_src/tools/attention.py:4
Functionbatchnorm_backward
(dout, cache)
paper-code/batch_normalization.py:29
Functionbatchnorm_forward
(x, gamma, beta, eps)
paper-code/batch_normalization.py:4
Methodcall
(self, v, k, q, mask=None)
paper-code/tensorflow_src/tools/attention.py:119
Methodcall
(self, attention_weights_cat)
paper-code/tensorflow_src/tools/attention.py:201
Methodcall
(self, query, values)
paper-code/tensorflow_src/models/seq2seq.py:11
Methodcall
(self, x, hidden)
paper-code/tensorflow_src/models/seq2seq.py:39
Methodcall
(self, x, hidden, enc_output)
paper-code/tensorflow_src/models/seq2seq.py:67
Methodcall
(self, inputs)
paper-code/tensorflow_src/models/transformer.py:32
Methodcall
(self, v, k, q, mask=None)
paper-code/tensorflow_src/models/InferSent.py:259
Methodclick_item_action
(self, point: QModelIndex)
search_kits.py:254
Functionconv2d_cosnorm
conv2d_cosnorm Usage : x : An input Tensor with shape [batch, height, width, channel] w : A convolutional kernel Tensor with
paper-code/conv2d_cosnorm.py:3
Functionconvert_delex
系统回复槽位生成,将结果保存在一个文件中
paper-code/tensorflow_src/models/task/common/pre_treat.py:98
Functioncreat_look_ahead_mask
(inputs)
paper-code/tensorflow_src/models/gpt2.py:14
Functioncreate_look_ahead_mask
(input)
paper-code/tensorflow_src/models/transformer.py:46
Functioncreate_padding_mask
用于创建输入序列的扩充部分的mask :param seq: 输入序列 :return: mask
paper-code/tensorflow_src/models/InferSent.py:311
Functiondecoder
(num_layers, num_heads, units, deep, dropout)
paper-code/tensorflow_src/models/gpt2.py:111
Methodforward
(self, inputs: torch.Tensor)
paper-code/pytorch_src/model/seq2seq.py:21
Methodforward
(self, query: torch.Tensor, values: torch.Tensor)
paper-code/pytorch_src/model/seq2seq.py:38
Methodforward
(self, inputs: torch.Tensor, hidden: torch.Tensor, enc_output: torch.Tensor)
paper-code/pytorch_src/model/seq2seq.py:71
Functiongenerator
()
paper-code/tensorflow_src/tools/preprocess_tfrecord.py:78
Methodget_config
(self)
paper-code/tensorflow_src/models/informer.py:17
Methodget_vocabs
获取对话数据集中的token集合,分为user和system两个token集合 :return: user和system两个token集合
paper-code/tensorflow_src/models/task/common/data_utils.py:268
Functiongpt2
(vocab_size, num_layers, units, deep, num_heads, dropout)
paper-code/tensorflow_src/models/gpt2.py:140
Functioninfer_sent
短文本匹配模型 :param vocab_size: token大小 :param num_layers: 编码解码的数量 :param units: 单元大小 :param embedding_dim: 词嵌入维度 :param num_heads: 多头
paper-code/tensorflow_src/models/InferSent.py:9
Functioninform_slot_tracker
informable插槽跟踪器,informable插槽是用户告知系统的信息,用 来约束对话的一些条件,系统为了完成任务必须满足这些条件 用来获得时间t的状态的槽值分布,比如price=cheap 输入为状态跟踪器的输入'state_t',输出为槽值分布'P(v_s
paper-code/tensorflow_src/models/task/model/tracker.py:4
Functioninformer
(embedding_dim: Any, enc_num_layers: Any, dec_num_layers: Any, batch_size: Any, num_heads: Any, dropout: Any,
paper-code/tensorflow_src/models/informer.py:399
Methodinitialize_hidden_state
(self)
paper-code/tensorflow_src/models/seq2seq.py:44
Functionload_data
数据加载方法,主要将分词好的数据进行整理,过程中保存字典文件,方便后续其他功能 使用,方法返回处理好的dataset,steps_per_epoch,checkpoint_prefix Args: dict_fn: 将训练数据的字典保存,用于以后使用,路径
paper-code/pytorch_src/common/data_utils.py:74
Functionload_data
加载对原始数据、本体数据、database数据处理好的数据集 :param dialogues_train: 原始对话数据路径 :param kb_fn: database数据路径 :param ontology_fn: 本体数据路径 :param toke
paper-code/tensorflow_src/models/task/common/data_utils.py:372
Functionload_dataset
获取Dataset :param record_path: :param batch_size: batch大小 :param buffer_size: 缓冲大小 :param num_parallel_reads: 读取线程数 :param data_t
paper-code/tensorflow_src/tools/preprocess_tfrecord.py:155
Functionload_dataset
数据加载方法,含四个元素的元组,包括如下: :return:input_tensor, input_token, target_tensor, target_token
paper-code/tensorflow_src/models/task/common/data_utils.py:88
Functionload_token_dict
加载字典方法 :return:input_token, target_token
paper-code/pytorch_src/common/data_utils.py:125
← previousnext →101–200 of 229, ranked by callers