In Memory Pred Dataset
In-memory prediction dataset.
InMemoryPredDataset
Bases: Dataset
Simple prediction dataset returning images along the sample axis.
Parameters:
| Name | Type | Description | Default |
|---|---|---|---|
prediction_config
|
InferenceConfig
|
Prediction configuration. |
required |
inputs
|
NDArray
|
Input data. |
required |