SampleMovieLensParams Struct Reference

The SampleMovieLensParams structure groups the additional parameters required by the MovieLens sample. More...

Inheritance diagram for SampleMovieLensParams:

Collaboration diagram for SampleMovieLensParams:

Public Attributes
int32_t	embeddingVecSize
int32_t	numUsers
int32_t	topKMovies
int32_t	numMoviesPerUser
std::string	ratingInputFile
bool	strict
std::map< int32_t, std::vector< int32_t > >	userToItemsMap
std::map< int32_t, std::vector< std::pair< int32_t, float > > >	userToExpectedItemProbMap
std::vector< OutputParams >	outParamsVec
std::string	uffFileName
	Filename of uff file of a network. More...
int32_t	batchSize {1}
	Number of inputs in a batch. More...
int32_t	dlaCore {-1}
	Specify the DLA core to run network on. More...
bool	int8 {false}
	Allow runnning the network in Int8 mode. More...
bool	fp16 {false}
	Allow running the network in FP16 mode. More...
std::vector< std::string >	dataDirs
	Directory paths where sample data files are stored. More...
std::vector< std::string >	inputTensorNames
std::vector< std::string >	outputTensorNames

Detailed Description

The SampleMovieLensParams structure groups the additional parameters required by the MovieLens sample.

Member Data Documentation

embeddingVecSize

int32_t SampleMovieLensParams::embeddingVecSize

numUsers

int32_t SampleMovieLensParams::numUsers

topKMovies

int32_t SampleMovieLensParams::topKMovies

numMoviesPerUser

int32_t SampleMovieLensParams::numMoviesPerUser

ratingInputFile

std::string SampleMovieLensParams::ratingInputFile

strict

bool SampleMovieLensParams::strict

userToItemsMap

std::map<int32_t, std::vector<int32_t> > SampleMovieLensParams::userToItemsMap

userToExpectedItemProbMap

std::map<int32_t, std::vector<std::pair<int32_t, float> > > SampleMovieLensParams::userToExpectedItemProbMap

outParamsVec

std::vector<OutputParams> SampleMovieLensParams::outParamsVec

uffFileName

std::string samplesCommon::UffSampleParams::uffFileName

inherited

Filename of uff file of a network.

batchSize

int32_t samplesCommon::SampleParams::batchSize {1}

inherited

Number of inputs in a batch.

dlaCore

int32_t samplesCommon::SampleParams::dlaCore {-1}

inherited

Specify the DLA core to run network on.

int8

bool samplesCommon::SampleParams::int8 {false}

inherited

Allow runnning the network in Int8 mode.

fp16

bool samplesCommon::SampleParams::fp16 {false}

inherited

Allow running the network in FP16 mode.

dataDirs

std::vector<std::string> samplesCommon::SampleParams::dataDirs

inherited

Directory paths where sample data files are stored.

inputTensorNames

std::vector<std::string> samplesCommon::SampleParams::inputTensorNames

inherited

outputTensorNames

std::vector<std::string> samplesCommon::SampleParams::outputTensorNames

inherited

---

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

• sampleMovieLens.cpp