package com.mooctest.demo;

import com.hankcs.hanlp.mining.word2vec.DocVectorModel;
import com.hankcs.hanlp.mining.word2vec.Word2VecTrainer;
import com.hankcs.hanlp.mining.word2vec.WordVectorModel;

import java.io.IOException;

public class DemoTrain {
    public static void main(String[] args) throws IOException {
//        Word2VecTrainer trainerBuilder = new Word2VecTrainer();
//        WordVectorModel wordVectorModel = trainerBuilder.train("/Users/major/Downloads/hanlp-wiki-vec-zh/hanlp-wiki-vec-zh.txt", "/Users/major/Downloads/hanlp-wiki-vec-zh/msr_vectors.txt");
//        wordVectorModel.nearest("中国");

//        WordVectorModel wordVectorModel = new WordVectorModel("/Users/major/Downloads/hanlp-wiki-vec-zh/hanlp-wiki-vec-zh.txt");
//        System.out.println(wordVectorModel.nearest("奖励"));

//        DocVectorModel docVectorModel1 = new DocVectorModel(new WordVectorModel("/Users/major/Downloads/hanlp-wiki-vec-zh/hanlp-wiki-vec-zh.txt"));
        DocVectorModel docVectorModel2 = new DocVectorModel(new WordVectorModel("/Users/major/Downloads/sgns.wiki.word"));
//        System.out.println(docVectorModel1.similarity("山西副省长贪污腐败开庭", "陕西村干部受贿违纪"));
        System.out.println(docVectorModel2.similarity("“全部”界面中可供选择的完整路径太少，只有部分路径存在（周边，亲子游，城市，1-3天，101-500，4-6月），绝大部分路径都不存在。", "在热门搜索-品类-全部筛选-滑到下面进行“自定义预计花费”，在该输入框中一单输入数字，页面立刻跳到上面，给用户输入造成困难，我用的雷电模拟器"));
    }
}
