public class WordConverter
extends java.lang.Object
Constructor and Description |
---|
WordConverter(java.util.List<java.lang.String> sentences,
Word2Vec vec) |
Modifier and Type | Method and Description |
---|---|
org.nd4j.linalg.api.ndarray.INDArray |
toInputMatrix() |
static org.nd4j.linalg.api.ndarray.INDArray |
toInputMatrix(java.util.List<Window> windows,
Word2Vec vec) |
org.nd4j.linalg.api.ndarray.INDArray |
toLabelMatrix(java.util.List<java.lang.String> labels) |
static org.nd4j.linalg.api.ndarray.INDArray |
toLabelMatrix(java.util.List<java.lang.String> labels,
java.util.List<Window> windows) |
public WordConverter(java.util.List<java.lang.String> sentences, Word2Vec vec)
public static org.nd4j.linalg.api.ndarray.INDArray toInputMatrix(java.util.List<Window> windows, Word2Vec vec)
public org.nd4j.linalg.api.ndarray.INDArray toInputMatrix()
public static org.nd4j.linalg.api.ndarray.INDArray toLabelMatrix(java.util.List<java.lang.String> labels, java.util.List<Window> windows)
public org.nd4j.linalg.api.ndarray.INDArray toLabelMatrix(java.util.List<java.lang.String> labels)