Package org.jetbrains.kotlinx.dl.dataset

Types

DataBatch
Link copied to clipboard
data class DataBatch
Content copied to clipboard

This class represents the batch of data in the Dataset.

DataLoader
Link copied to clipboard
interface DataLoader<D>
Content copied to clipboard

And interface for loading data from the provided data source.

Dataset
Link copied to clipboard
abstract class Dataset
Content copied to clipboard

Just abstract Dataset.

OnFlyImageDataset
Link copied to clipboard
class OnFlyImageDataset<D> : Dataset
Content copied to clipboard

This dataset keeps all data on disk and generates batches on the fly using provided dataLoader.

OnHeapDataset
Link copied to clipboard
class OnHeapDataset : Dataset
Content copied to clipboard

Basic class to handle features x and labels y.

Functions

buildStringRepr
Link copied to clipboard
fun OnHeapDataset.buildStringRepr(xString: String, yString: String): String
Content copied to clipboard

Builds intermediate OnHeapDataset string representation.

evaluate
Link copied to clipboard
fun InferenceModel.evaluate(dataset: Dataset, metric: Metrics): Double
Content copied to clipboard

Evaluates dataset via metric.

fullToString
Link copied to clipboard
fun OnHeapDataset.fullToString(): String
Content copied to clipboard

Creates OnHeapDataset string representation for full of data.

predict
Link copied to clipboard
fun InferenceModel.predict(dataset: Dataset): List<Int>
Content copied to clipboard

Predicts labels for all observation in dataset.

toString
Link copied to clipboard
fun OnHeapDataset.toString(): String
Content copied to clipboard

Creates OnHeapDataset string representation for part of data.