cerebras.modelzoo.data_preparation.nlp.slimpajama.preprocessing.datasets.RedPajamaReplication

class cerebras.modelzoo.data_preparation.nlp.slimpajama.preprocessing.datasets.RedPajamaReplication

Bases: cerebras.modelzoo.data_preparation.nlp.slimpajama.preprocessing.datasets.Dataset

Methods

already_shuffled	Datasets where the source is already shuffled should override this to return True so that it isn't shuffled again.
dir_path	Path to the directory
documents	A generator producing all documents in the dataset.
name	Human-readable name of tfhe dataset
num_docs	Return an estimate of the dataset number of documents.
sample_documents
short_documents_path	Path to the file with short documents
size	Return an estimate of the dataset size.

__init__(datasets, duplicates, short_docs)

name()

Human-readable name of tfhe dataset

size()

Return an estimate of the dataset size. Implementations may use a faster, less accurate estimate.

num_docs()

Return an estimate of the dataset number of documents. Implementations may use a faster, less accurate estimate.

documents(queues)

A generator producing all documents in the dataset.

already_shuffled()

Datasets where the source is already shuffled should override this to return True so that it isn’t shuffled again.

dir_path()

Path to the directory

short_documents_path()

Path to the file with short documents

previous

cerebras.modelzoo.data_preparation.nlp.slimpajama.preprocessing.datasets.RedPajamaGithubDataset

next

cerebras.modelzoo.data_preparation.nlp.slimpajama.preprocessing.datasets.RedPajamaStackExchangeDataset