Package org.deeplearning4j.text.movingwindow

Class WordConverter

java.lang.Object

org.deeplearning4j.text.movingwindow.WordConverter

public class WordConverter
extends Object

Constructor Summary

Constructors

Constructor	Description
WordConverter(List<String> sentences, Word2Vec vec)

Method Summary

Modifier and Type	Method	Description
org.nd4j.linalg.api.ndarray.INDArray	toInputMatrix()
static org.nd4j.linalg.api.ndarray.INDArray	toInputMatrix(List<Window> windows, Word2Vec vec)
org.nd4j.linalg.api.ndarray.INDArray	toLabelMatrix(List<String> labels)
static org.nd4j.linalg.api.ndarray.INDArray	toLabelMatrix(List<String> labels, List<Window> windows)

Methods inherited from class java.lang.Object

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

Constructor Detail

WordConverter

public WordConverter(List<String> sentences,
                     Word2Vec vec)

Method Detail

toInputMatrix

public static org.nd4j.linalg.api.ndarray.INDArray toInputMatrix(List<Window> windows,
                                                                 Word2Vec vec)

toInputMatrix

public org.nd4j.linalg.api.ndarray.INDArray toInputMatrix()

toLabelMatrix

public static org.nd4j.linalg.api.ndarray.INDArray toLabelMatrix(List<String> labels,
                                                                 List<Window> windows)

toLabelMatrix

public org.nd4j.linalg.api.ndarray.INDArray toLabelMatrix(List<String> labels)