Class TFRecordDataset

Inheritance Relationships

Base Type

Class Documentation

class TFRecordDataset : public mindspore::dataset::Dataset

Public Functions

TFRecordDataset(const std::vector<std::vector<char>> &dataset_files, std::shared_ptr<SchemaObj> schema, const std::vector<std::vector<char>> &columns_list, int64_t num_samples, ShuffleMode shuffle, int32_t num_shards, int32_t shard_id, bool shard_equal_rows, std::shared_ptr<DatasetCache> cache)

Constructor.

Note

Parameter ‘schema’ is shared pointer to Schema object