1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
| package com.walker.embedding;
|
| /**
| * 向量生成器。
| * @author 时克英
| * @date 2022-08-23
| */
| public interface VectorGenerator {
|
| /**
| * 初始化加载词库与向量模型。
| * @param wordFile 词库文件路径，如：d:/demo/dict.txt
| * @param embeddingFile 向量维度模型路径，如：d:/demo/w2v_sogou_dim300.npy
| */
| void initLoadDict(String wordFile, String embeddingFile);
|
| /**
| * 返回词向量。
| * @param word 给定单词
| * @return
| */
| float[] getWordEmbedding(String word);
|
| /**
| * 返回两个词向量余玄相似度。
| * @param feature1
| * @param feature2
| * @return
| */
| float cosineSim(float[] feature1, float[] feature2);
| }
|
|
