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

context

reflection context

---

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

• Gems/MachineLearning/Code/Source/Assets/MnistDataLoader.h