public class TimeSeriesUtils extends Object
Modifier and Type | Method and Description |
---|---|
static org.nd4j.linalg.api.ndarray.INDArray |
movingAverage(org.nd4j.linalg.api.ndarray.INDArray toAvg,
int n)
Calculate a moving average given the length
|
static org.nd4j.linalg.primitives.Pair<org.nd4j.linalg.api.ndarray.INDArray,int[]> |
pullLastTimeSteps(org.nd4j.linalg.api.ndarray.INDArray pullFrom,
org.nd4j.linalg.api.ndarray.INDArray mask)
Extract out the last time steps (2d array from 3d array input) accounting for the mask layer, if present.
|
static org.nd4j.linalg.api.ndarray.INDArray |
reshape2dTo3d(org.nd4j.linalg.api.ndarray.INDArray in,
int miniBatchSize) |
static org.nd4j.linalg.api.ndarray.INDArray |
reshape3dTo2d(org.nd4j.linalg.api.ndarray.INDArray in) |
static org.nd4j.linalg.api.ndarray.INDArray |
reshapePerOutputTimeSeriesMaskTo2d(org.nd4j.linalg.api.ndarray.INDArray perOutputTimeSeriesMask) |
static org.nd4j.linalg.api.ndarray.INDArray |
reshapeTimeSeriesMaskToVector(org.nd4j.linalg.api.ndarray.INDArray timeSeriesMask)
Reshape time series mask arrays.
|
static org.nd4j.linalg.api.ndarray.INDArray |
reshapeVectorToTimeSeriesMask(org.nd4j.linalg.api.ndarray.INDArray timeSeriesMaskAsVector,
int minibatchSize)
Reshape time series mask arrays.
|
static org.nd4j.linalg.api.ndarray.INDArray |
reverseTimeSeries(org.nd4j.linalg.api.ndarray.INDArray in)
Reverse an input time series along the time dimension
|
static org.nd4j.linalg.api.ndarray.INDArray |
reverseTimeSeriesMask(org.nd4j.linalg.api.ndarray.INDArray mask)
Reverse a (per time step) time series mask, with shape [minibatch, timeSeriesLength]
|
public static org.nd4j.linalg.api.ndarray.INDArray movingAverage(org.nd4j.linalg.api.ndarray.INDArray toAvg, int n)
toAvg
- the array to averagen
- the length of the moving windowpublic static org.nd4j.linalg.api.ndarray.INDArray reshapeTimeSeriesMaskToVector(org.nd4j.linalg.api.ndarray.INDArray timeSeriesMask)
timeSeriesMask
- Mask array to reshape to a column vectorpublic static org.nd4j.linalg.api.ndarray.INDArray reshapeVectorToTimeSeriesMask(org.nd4j.linalg.api.ndarray.INDArray timeSeriesMaskAsVector, int minibatchSize)
timeSeriesMaskAsVector
- Mask array to reshape to a column vectorpublic static org.nd4j.linalg.api.ndarray.INDArray reshapePerOutputTimeSeriesMaskTo2d(org.nd4j.linalg.api.ndarray.INDArray perOutputTimeSeriesMask)
public static org.nd4j.linalg.api.ndarray.INDArray reshape3dTo2d(org.nd4j.linalg.api.ndarray.INDArray in)
public static org.nd4j.linalg.api.ndarray.INDArray reshape2dTo3d(org.nd4j.linalg.api.ndarray.INDArray in, int miniBatchSize)
public static org.nd4j.linalg.api.ndarray.INDArray reverseTimeSeries(org.nd4j.linalg.api.ndarray.INDArray in)
in
- Input activations to reverse, with shape [minibatch, size, timeSeriesLength]public static org.nd4j.linalg.api.ndarray.INDArray reverseTimeSeriesMask(org.nd4j.linalg.api.ndarray.INDArray mask)
mask
- Mask to reverse along time dimensionpublic static org.nd4j.linalg.primitives.Pair<org.nd4j.linalg.api.ndarray.INDArray,int[]> pullLastTimeSteps(org.nd4j.linalg.api.ndarray.INDArray pullFrom, org.nd4j.linalg.api.ndarray.INDArray mask)
pullFrom
- Input time series array (rank 3) to pull the last time steps frommask
- Mask array (rank 2). May be nullCopyright © 2018. All rights reserved.