Word2vec 案例应用

接口：模型训练 public static void saveModel(String input, String output) { Learn learn = new Learn(); learn.learnFile(new File("G:/output_bd_2.txt")); learn.saveModel(new File("G:/vectors_gs_bd_2.bin")); } 说明：搜索结果输出 input：输入所要训练的文本； output：输出所要训练的模型； vectors_gs_bd.bin：模型文件 public static Set<WordEntry> semantic(String input) { Word2VEC vec2 = new Word2VEC(); vec2.loadJavaModel("./vectors_gs_bd.bin") ; return vec2.distance(input); } 说明：搜索结果输出 input：输入所要查询的字符串；