org.deeplearning4j.models.word2vec.iterator

Class Word2VecDataFetcher

java.lang.Object

org.deeplearning4j.models.word2vec.iterator.Word2VecDataFetcher

All Implemented Interfaces:

Serializable, org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher

public class Word2VecDataFetcher
extends Object
implements org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
Word2VecDataFetcher(String path, Word2Vec vec, List<String> labels)

Method Summary

Modifier and Type	Method and Description
int	cursor()
void	fetch(int numExamples)
int	getBatch()
static Pattern	getBegin()
List<Window>	getCache()
static Pattern	getEnd()
Iterator<File>	getFiles()
List<String>	getLabels()
Word2Vec	getVec()
boolean	hasMore()
int	inputColumns()
org.nd4j.linalg.dataset.DataSet	next()
void	reset()
int	totalExamples()
int	totalOutcomes()

Methods inherited from class java.lang.Object

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

Constructor Detail

Word2VecDataFetcher

public Word2VecDataFetcher(String path,
                           Word2Vec vec,
                           List<String> labels)

Method Detail

next

public org.nd4j.linalg.dataset.DataSet next()

Specified by:

next in interface org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher

totalExamples

public int totalExamples()

Specified by:

totalExamples in interface org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher

inputColumns

public int inputColumns()

Specified by:

inputColumns in interface org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher

totalOutcomes

public int totalOutcomes()

Specified by:

totalOutcomes in interface org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher

reset

public void reset()

Specified by:

reset in interface org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher

cursor

public int cursor()

Specified by:

cursor in interface org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher

hasMore

public boolean hasMore()

Specified by:

hasMore in interface org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher

fetch

public void fetch(int numExamples)

Specified by:

fetch in interface org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher

getFiles

public Iterator<File> getFiles()

getVec

public Word2Vec getVec()

getBegin

public static Pattern getBegin()

getEnd

public static Pattern getEnd()

getLabels

public List<String> getLabels()

getBatch

public int getBatch()

getCache

public List<Window> getCache()