com.hankcs.hanlp.mining.word2vec

Class DocVectorModel

java.lang.Object

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

com.hankcs.hanlp.mining.word2vec.DocVectorModel

public class DocVectorModel
extends AbstractVectorModel<Integer>

文档向量模型

Author:

hankcs

Constructor Summary

Constructors

Constructor and Description
DocVectorModel(WordVectorModel wordVectorModel)
DocVectorModel(WordVectorModel wordVectorModel, Segment segment, boolean filter)

Method Summary

Modifier and Type	Method and Description
Vector	addDocument(int id, String content) 添加文档
int	dimension() 模型中的词向量维度
void	enableFilter(boolean filter) 激活/关闭停用词过滤器
Segment	getSegment()
boolean	isFilterEnabled() 是否激活了停用词过滤器
List<Map.Entry<Integer,Float>>	nearest(String query) 查询最相似的前10个文档
List<Map.Entry<Integer,Float>>	nearest(String query, int n) 查询最相似的前n个文档
Vector	query(String content) 将一个文档转为向量
void	setSegment(Segment segment)
float	similarity(String what, String with) 文档相似度计算

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

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

DocVectorModel

public DocVectorModel(WordVectorModel wordVectorModel)

DocVectorModel

public DocVectorModel(WordVectorModel wordVectorModel,
                      Segment segment,
                      boolean filter)

Method Detail

addDocument

public Vector addDocument(int id,
                          String content)

添加文档

Parameters:

id - 文档id

content - 文档内容

Returns:

文档向量

nearest

public List<Map.Entry<Integer,Float>> nearest(String query)

查询最相似的前10个文档

Parameters:

query - 查询语句（或者说一个文档的内容）

Returns:

nearest

public List<Map.Entry<Integer,Float>> nearest(String query,
                                              int n)

查询最相似的前n个文档

Parameters:

query - 查询语句（或者说一个文档的内容）

Returns:

query

public Vector query(String content)

将一个文档转为向量

Specified by:

query in class AbstractVectorModel<Integer>

Parameters:

content - 文档

Returns:

向量

dimension

public int dimension()

Description copied from class: AbstractVectorModel

模型中的词向量维度

Overrides:

dimension in class AbstractVectorModel<Integer>

Returns:

similarity

public float similarity(String what,
                        String with)

文档相似度计算

Parameters:

what -

with -

Returns:

getSegment

public Segment getSegment()

setSegment

public void setSegment(Segment segment)

isFilterEnabled

public boolean isFilterEnabled()

是否激活了停用词过滤器

Returns:

enableFilter

public void enableFilter(boolean filter)

激活/关闭停用词过滤器

Parameters:

filter -