com.hankcs.hanlp.mining.word2vec

Class WordVectorModel

java.lang.Object

com.hankcs.hanlp.mining.word2vec.AbstractVectorModel<String>

com.hankcs.hanlp.mining.word2vec.WordVectorModel

public class WordVectorModel
extends AbstractVectorModel<String>

词向量模型

Author:

hankcs

Constructor Summary

Constructors

Constructor and Description
WordVectorModel(String modelFileName) 加载模型
WordVectorModel(String modelFileName, Map<String,Vector> storage) 加载模型

Method Summary

Modifier and Type	Method and Description
List<Map.Entry<String,Float>>	analogy(String A, String B, String C) 返回跟 A - B + C 最相似的词语,比如 中国 - 北京 + 东京 = 日本。输入顺序按照 中国 北京 东京
List<Map.Entry<String,Float>>	analogy(String A, String B, String C, int size) 返回跟 A - B + C 最相似的词语,比如 中国 - 北京 + 东京 = 日本。输入顺序按照 中国 北京 东京
Vector	query(String query) 查询抽象文本对应的向量。此方法应当保证返回单位向量。

Methods inherited from class com.hankcs.hanlp.mining.word2vec.AbstractVectorModel

dimension, nearest, nearest, nearest, nearest, remove, similarity, size, vector

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

WordVectorModel

public WordVectorModel(String modelFileName)
                throws IOException

加载模型

Parameters:

modelFileName - 模型路径

Throws:

IOException - 加载错误

WordVectorModel

public WordVectorModel(String modelFileName,
                       Map<String,Vector> storage)
                throws IOException

加载模型

Parameters:

modelFileName - 模型路径

storage - 一个空白的Map（HashMap等）

Throws:

IOException - 加载错误

Method Detail

analogy

public List<Map.Entry<String,Float>> analogy(String A,
                                             String B,
                                             String C)

返回跟 A - B + C 最相似的词语,比如 中国 - 北京 + 东京 = 日本。输入顺序按照 中国 北京 东京

Parameters:

A - 做加法的词语

B - 做减法的词语

C - 做加法的词语

Returns:

与(A - B + C)语义距离最近的词语及其相似度列表

analogy

public List<Map.Entry<String,Float>> analogy(String A,
                                             String B,
                                             String C,
                                             int size)

返回跟 A - B + C 最相似的词语,比如 中国 - 北京 + 东京 = 日本。输入顺序按照 中国 北京 东京

Parameters:

A - 做加法的词语

B - 做减法的词语

C - 做加法的词语

size - topN个

Returns:

与(A - B + C)语义距离最近的词语及其相似度列表

query

public Vector query(String query)

Description copied from class: AbstractVectorModel

查询抽象文本对应的向量。此方法应当保证返回单位向量。

Specified by:

query in class AbstractVectorModel<String>

Returns: