public interface MultiDataSet extends Serializable
Modifier and Type | Method and Description |
---|---|
List<MultiDataSet> |
asList()
Split the MultiDataSet into a list of individual examples.
|
MultiDataSet |
copy()
Clone the dataset
|
List<Serializable> |
getExampleMetaData()
Get the example metadata, or null if no metadata has been set
|
<T extends Serializable> |
getExampleMetaData(Class<T> metaDataType)
Get the example metadata, or null if no metadata has been set
Note: this method results in an unchecked cast - care should be taken when using this! |
INDArray[] |
getFeatures()
Get all of the input features, as an array of INDArrays
|
INDArray |
getFeatures(int index)
Get a single feature/input array
|
INDArray |
getFeaturesMaskArray(int index)
Get the specified feature mask array.
|
INDArray[] |
getFeaturesMaskArrays()
Get the feature mask arrays.
|
INDArray[] |
getLabels()
Get all of the labels, as an array of INDArrays
|
INDArray |
getLabels(int index)
Get a single label/output array
|
INDArray |
getLabelsMaskArray(int index)
Get the specified label mask array.
|
INDArray[] |
getLabelsMaskArrays()
Get the labels mask arrays.
|
boolean |
hasMaskArrays()
Whether there are any mask arrays (features or labels) present for this MultiDataSet
|
void |
load(File from)
Load the contents of this MultiDataSet from the specified file
|
void |
load(InputStream from)
Load the contents of this MultiDataSet from the specified stream.
|
int |
numFeatureArrays()
Number of arrays of features/input data in the MultiDataSet
|
int |
numLabelsArrays()
Number of arrays of label/output data in the MultiDataSet
|
void |
save(File to)
Save this MultiDataSet to the specified file
|
void |
save(OutputStream to)
Save this MultiDataSet to the specified stream.
|
void |
setExampleMetaData(List<? extends Serializable> exampleMetaData)
Set the metadata for this MultiDataSet
By convention: the metadata can be any serializable object, one per example in the MultiDataSet |
void |
setFeatures(INDArray[] features)
Set all of the features arrays for the MultiDataSet
|
void |
setFeatures(int idx,
INDArray features)
Set a single features array (by index) for the MultiDataSet
|
void |
setFeaturesMaskArray(int idx,
INDArray maskArray)
Set a single feature mask array by index
|
void |
setFeaturesMaskArrays(INDArray[] maskArrays)
Set the feature mask arrays
|
void |
setLabels(INDArray[] labels)
Set all of the labels arrays for the MultiDataSet
|
void |
setLabels(int idx,
INDArray labels)
Set a single labels array (by index) for the MultiDataSet
|
void |
setLabelsMaskArray(INDArray[] labels)
Set the labels mask arrays
|
void |
setLabelsMaskArray(int idx,
INDArray labelsMaskArray)
Set a single labels mask array by index
|
int numFeatureArrays()
int numLabelsArrays()
INDArray[] getFeatures()
INDArray getFeatures(int index)
void setFeatures(INDArray[] features)
void setFeatures(int idx, INDArray features)
INDArray[] getLabels()
INDArray getLabels(int index)
void setLabels(INDArray[] labels)
void setLabels(int idx, INDArray labels)
boolean hasMaskArrays()
INDArray[] getFeaturesMaskArrays()
INDArray getFeaturesMaskArray(int index)
void setFeaturesMaskArrays(INDArray[] maskArrays)
void setFeaturesMaskArray(int idx, INDArray maskArray)
INDArray[] getLabelsMaskArrays()
INDArray getLabelsMaskArray(int index)
void setLabelsMaskArray(INDArray[] labels)
void setLabelsMaskArray(int idx, INDArray labelsMaskArray)
void save(OutputStream to) throws IOException
IOException
void save(File to) throws IOException
IOException
void load(InputStream from) throws IOException
IOException
void load(File from) throws IOException
IOException
List<MultiDataSet> asList()
MultiDataSet copy()
void setExampleMetaData(List<? extends Serializable> exampleMetaData)
exampleMetaData
- Example metadata to set<T extends Serializable> List<T> getExampleMetaData(Class<T> metaDataType)
T
- Type of metadatametaDataType
- Class of the metadata (used for type information)List<Serializable> getExampleMetaData()
#getExampleMetaData(Class)} for convenience method for types
Copyright © 2016. All Rights Reserved.