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);
}