Doc2VecUtil.java 943 B

1234567891011121314151617181920212223242526272829
  1. package com.mooctest.util;
  2. import com.hankcs.hanlp.mining.word2vec.DocVectorModel;
  3. import com.hankcs.hanlp.mining.word2vec.WordVectorModel;
  4. import java.io.IOException;
  5. public class Doc2VecUtil {
  6. private static volatile DocVectorModel docVectorModel;
  7. public static DocVectorModel loadModel() {
  8. if (docVectorModel == null) {
  9. synchronized (Doc2VecUtil.class) {
  10. if (docVectorModel == null) {
  11. try {
  12. docVectorModel = new DocVectorModel(new WordVectorModel("C:\\Users\\WRZ\\Desktop\\sgns.wiki.word"));
  13. // docVectorModel = new DocVectorModel(new WordVectorModel("/Users/insomnialee/Desktop/sgns.wiki.word"));
  14. } catch (IOException e) {
  15. e.printStackTrace();
  16. }
  17. }
  18. }
  19. }
  20. return docVectorModel;
  21. }
  22. }