Open 3D Engine MachineLearning Gem API Reference 25.10.0
O3DE is an open-source, fully-featured, high-fidelity, modular 3D engine for building games and simulations, available to every industry.
Classes | Public Member Functions | Static Public Member Functions | List of all members
MachineLearning::MnistDataLoader Class Reference

#include <MnistDataLoader.h>

Inherits MachineLearning::ILabeledTrainingData.

Public Member Functions

 AZ_TYPE_INFO (MnistDataLoader, "{3F4C0F29-4E7E-4CAF-A331-EAC3D2D9409E}", ILabeledTrainingData)
 
bool LoadArchive (const AZ::IO::Path &imageFilename, const AZ::IO::Path &labelFilename) override
 
AZStd::size_t GetSampleCount () const override
 Returns the total number of samples contained in the training data set.
 
const AZ::VectorN & GetLabelByIndex (AZStd::size_t index) override
 Returns the index-th label in the training data set.
 
const AZ::VectorN & GetDataByIndex (AZStd::size_t index) override
 Returns the index-th set of activations in the training data set.
 
- Public Member Functions inherited from MachineLearning::ILabeledTrainingData
 AZ_TYPE_INFO (ILabeledTrainingData, "{50DF457E-3EAC-4114-8444-023E64973AD9}")
 

Static Public Member Functions

static void Reflect (AZ::ReflectContext *context)
 

Detailed Description

A class that can load the MNIST training data set. https://en.wikipedia.org/wiki/MNIST_database

Member Function Documentation

◆ GetDataByIndex()

const AZ::VectorN & MachineLearning::MnistDataLoader::GetDataByIndex ( AZStd::size_t  index)
overridevirtual

Returns the index-th set of activations in the training data set.

Implements MachineLearning::ILabeledTrainingData.

◆ GetLabelByIndex()

const AZ::VectorN & MachineLearning::MnistDataLoader::GetLabelByIndex ( AZStd::size_t  index)
overridevirtual

Returns the index-th label in the training data set.

Implements MachineLearning::ILabeledTrainingData.

◆ GetSampleCount()

AZStd::size_t MachineLearning::MnistDataLoader::GetSampleCount ( ) const
overridevirtual

Returns the total number of samples contained in the training data set.

Implements MachineLearning::ILabeledTrainingData.

◆ LoadArchive()

bool MachineLearning::MnistDataLoader::LoadArchive ( const AZ::IO::Path &  imageFilename,
const AZ::IO::Path &  labelFilename 
)
overridevirtual

ILabeledTrainingData interface

Implements MachineLearning::ILabeledTrainingData.

◆ Reflect()

static void MachineLearning::MnistDataLoader::Reflect ( AZ::ReflectContext *  context)
static

AzCore Reflection.

Parameters
contextreflection context
The documentation for this class was generated from the following file: