Companion

object Companion

Functions

convertByteToFloat

fun convertByteToFloat(label: Byte): Float

Creates float label.

create

fun create(features: Array<FloatArray>, labels: FloatArray): OnHeapDataset

Takes data from external data features and labels to create dataset OnHeapDataset.

fun create(featuresGenerator: () -> Array<FloatArray>, labelGenerator: () -> FloatArray): OnHeapDataset

Takes the data from generators featuresGenerator and labelGenerator to create an OnHeapDataset.

fun create(pathToData: File, labels: FloatArray, preprocessing: Operation<BufferedImage, Pair<FloatArray, TensorShape>>): OnHeapDataset

Creates an OnHeapDataset from pathToData and labels using preprocessing to prepare images.

fun create(pathToData: File, labelGenerator: LabelGenerator<File>, preprocessing: Operation<BufferedImage, Pair<FloatArray, TensorShape>> = ConvertToFloatArray()): OnHeapDataset

Creates an OnHeapDataset from pathToData and labelGenerator with preprocessing to prepare images.

fun create(featuresPath: String, labelsPath: String, numClasses: Int, featuresExtractor: (String) -> Array<FloatArray>, labelExtractor: (String, Int) -> FloatArray): OnHeapDataset

Takes data located in featuresPath, labelsPath with numClasses, extracts data and labels via featuresExtractor and labelExtractor to create an OnHeapDataset.

createTrainAndTestDatasets

fun createTrainAndTestDatasets(trainFeaturesPath: String, trainLabelsPath: String, testFeaturesPath: String, testLabelsPath: String, numClasses: Int, featuresExtractor: (String) -> Array<FloatArray>, labelExtractor: (String, Int) -> FloatArray): Pair<OnHeapDataset, OnHeapDataset>

Takes data located in trainFeaturesPath, trainLabelsPath, testFeaturesPath, testLabelsPath with numClasses, extracts data and labels via featuresExtractor and labelExtractor to create a pair of OnHeapDataset for training and testing.

toOneHotVector

fun toOneHotVector(numClasses: Int, label: Byte): FloatArray

Creates binary vector with size numClasses from label.