[发明专利]一种基于LSTM和多样化集束搜索的多样化API推荐方法及装置

权利要求书

1.一种基于LSTM和多样化集束搜索的多样化API推荐方法，其特征在于，包括：

S1：获取原始数据集，对原始数据集中对数据进行分析，统计每个API出现的次数，使用IDF公式计算逆向文件频率，再基于逆向文件频率构建多样化得分，其中，多样化得分用以表示每个API的多样化程度；

S2：构建编码器，将自然语言API查询按单词划分为序列后，逐个输入编码器，将自然语言的API查询编码为基于时间步的编码器隐藏状态；

S3：构建解码器，解码器根据基于时间步的编码器隐藏状态并结合注意力机制，按时间步生成解码器隐藏状态，并使用多样化集束搜索进行多样化API序列的输出。

2.如权利要求1所述的多样化API推荐方法，其特征在于，S1具体包括：

S1.1：从获取的原始数据集中提取出所有样本的标签，标签为API序列信息；

S1.2：遍历每条API序列信息，从API序列中分离出单独的API；

S1.3：统计每个API出现的次数，过滤无实际意义的API；

S1.4：利用IDF公式计算出每个API的IDF系数，API的IDF系数代表API的流行程度，计算公式为：

其中，IDF_i表示第i个API的IDF系数，count(API_i)表示第i个API在数据集中出现的次数；

S1.5：根据步骤S1.4所计算的每个API的IDF系数以及预先设置的API的多样化强度权重，计算每个API的多样化得分，API的多样化得分代表每个API的多样化程度，计算公式为：

其中，表示第i个API的多样化得分，weight_API表示API的多样化强度权重。

3.如权利要求1所述的多样化API推荐方法，其特征在于，S2具体包括：

S2.1：给定自然语言查询Q,将其按单词划分为序列w₁,…,w_i,…,w_n，其中查询Q中的第i个单词为w_i，编码自然语言表达的查询单词，使用独热编码将单词w_i编码为高维稀疏的输入向量o_i；

S2.2：构建编码器，编码器包括输入层、隐藏层和输出层，输入层由一层embedding层构成，隐藏层由一层LSTM层所组成；

S2.3：通过编码器输入层embedding映射将高维稀疏的输入向量转化为定长的低维稠密向量：

x_i＝Embedding(o_i)

其中，x_i表示低维稠密向量，o_i表示输入向量；

S2.4：通过编码器的隐藏层按照时间步t根据前一个隐藏层状态和输入层的结果向量x_t计算出当前的隐藏层状态：

h_t＝LSTM(h_t-1,x_t)

其中，x_t表示输入层的结果向量，h_t-1表示时间步t的前一个时间步的隐藏层状态，h_t表示当前时间步t的隐藏层状态；

S2.5：通过编码器的输出层得到每个时间步t的编码器的隐藏状态。