Package org.nlpub.watset.util

Class Word2VecFormat

java.lang.Object

org.nlpub.watset.util.Word2VecFormat

public final class Word2VecFormat
extends Object

Utilities for handling the Word2Vec text format for node embeddings.

See Also:

• Mikolov (NIPS 2013)
• gensim: models.keyedvectors – Store and query word vectors
• word2vec: Tool for computing continuous distributed representations of words
• NodeEmbedding

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static enum	Word2VecFormat.SpaceStrategy	Space character handling strategy.

Field Summary

Fields

Modifier and Type	Field	Description
static final Pattern	SPACES	A pattern that matches a space.

Method Summary

Modifier and Type	Method	Description
static <V> void	write(BufferedWriter writer, List<NodeEmbedding<V>> embeddings, Word2VecFormat.SpaceStrategy spaceStrategy)	Write the file in the Word2Vec format representing the node embeddings.

Methods inherited from class java.lang.Object

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

Field Details

SPACES

public static final Pattern SPACES

A pattern that matches a space.

Method Details

write

public static <V> void write(BufferedWriter writer,
 List<NodeEmbedding<V>> embeddings,
 Word2VecFormat.SpaceStrategy spaceStrategy)
                      throws IOException

Write the file in the Word2Vec format representing the node embeddings.

Type Parameters:

V - the type of nodes in the graph

Parameters:

writer - the writer

embeddings - the embeddings

spaceStrategy - the space handling strategy

Throws:

IOException - if an I/O error occurs