public class Word2VecDataFetcher
extends java.lang.Object
implements org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher
Constructor and Description |
---|
Word2VecDataFetcher(java.lang.String path,
Word2Vec vec,
java.util.List<java.lang.String> labels) |
Modifier and Type | Method and Description |
---|---|
int |
cursor() |
void |
fetch(int numExamples) |
int |
getBatch() |
static java.util.regex.Pattern |
getBegin() |
java.util.List<Window> |
getCache() |
static java.util.regex.Pattern |
getEnd() |
java.util.Iterator<java.io.File> |
getFiles() |
java.util.List<java.lang.String> |
getLabels() |
Word2Vec |
getVec() |
boolean |
hasMore() |
int |
inputColumns() |
org.nd4j.linalg.dataset.DataSet |
next() |
void |
reset() |
int |
totalExamples() |
int |
totalOutcomes() |
public Word2VecDataFetcher(java.lang.String path, Word2Vec vec, java.util.List<java.lang.String> labels)
public org.nd4j.linalg.dataset.DataSet next()
next
in interface org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher
public int totalExamples()
totalExamples
in interface org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher
public int inputColumns()
inputColumns
in interface org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher
public int totalOutcomes()
totalOutcomes
in interface org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher
public void reset()
reset
in interface org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher
public int cursor()
cursor
in interface org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher
public boolean hasMore()
hasMore
in interface org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher
public void fetch(int numExamples)
fetch
in interface org.nd4j.linalg.dataset.api.iterator.fetcher.DataSetFetcher
public java.util.Iterator<java.io.File> getFiles()
public Word2Vec getVec()
public static java.util.regex.Pattern getBegin()
public static java.util.regex.Pattern getEnd()
public java.util.List<java.lang.String> getLabels()
public int getBatch()
public java.util.List<Window> getCache()