com.hankcs.hanlp.mining.word2vec

Class AbstractVectorModel<K>

java.lang.Object

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

Direct Known Subclasses:

DocVectorModel, WordVectorModel

public abstract class AbstractVectorModel<K>
extends Object

抽象的向量模型，将抽象的对象映射为向量

Author:

hankcs

Constructor Summary

Constructors

Constructor and Description
AbstractVectorModel()
AbstractVectorModel(Map<K,Vector> storage)

Method Summary

Modifier and Type	Method and Description
int	dimension() 模型中的词向量维度
List<Map.Entry<K,Float>>	nearest(K key) 查询与词语最相似的词语
List<Map.Entry<K,Float>>	nearest(K key, int size) 查询与key最相似的元素
List<Map.Entry<K,Float>>	nearest(Vector vector) 获取与向量最相似的词语（默认10个）
List<Map.Entry<K,Float>>	nearest(Vector vector, int size) 获取与向量最相似的词语
abstract Vector	query(String query) 查询抽象文本对应的向量。此方法应当保证返回单位向量。
Vector	remove(K key) 删除元素
float	similarity(K what, K with) 余弦相似度
int	size() 模型中的词向量总数（词表大小）
Vector	vector(K key) 获取一个键的向量（键不会被预处理）

Methods inherited from class java.lang.Object

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

Constructor Detail

AbstractVectorModel

public AbstractVectorModel(Map<K,Vector> storage)

AbstractVectorModel

public AbstractVectorModel()

Method Detail

vector

public final Vector vector(K key)

获取一个键的向量（键不会被预处理）

Parameters:

key - 键

Returns:

向量

similarity

public float similarity(K what,
                        K with)

余弦相似度

Parameters:

what - 一个词

with - 另一个词

Returns:

余弦相似度

nearest

public List<Map.Entry<K,Float>> nearest(K key,
                                        int size)

查询与key最相似的元素

Parameters:

key - 键

size - topN个

Returns:

键值对列表, 键是相似词语, 值是相似度, 按相似度降序排列

nearest

public List<Map.Entry<K,Float>> nearest(Vector vector,
                                        int size)

获取与向量最相似的词语

Parameters:

vector - 向量

size - topN个

Returns:

键值对列表, 键是相似词语, 值是相似度, 按相似度降序排列

nearest

public List<Map.Entry<K,Float>> nearest(Vector vector)

获取与向量最相似的词语（默认10个）

Parameters:

vector - 向量

Returns:

键值对列表, 键是相似词语, 值是相似度, 按相似度降序排列

nearest

public List<Map.Entry<K,Float>> nearest(K key)

查询与词语最相似的词语

Parameters:

key - 词语

Returns:

键值对列表, 键是相似词语, 值是相似度, 按相似度降序排列

query

public abstract Vector query(String query)

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

Parameters:

query -

Returns:

size

public int size()

模型中的词向量总数（词表大小）

Returns:

dimension

public int dimension()

模型中的词向量维度

Returns:

remove

public Vector remove(K key)

删除元素

Parameters:

key -

Returns: