modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaArXivDataset

class modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaArXivDataset

Bases: modelzoo.transformers.data_processing.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
num_duplicate_docs
num_short_docs
short_documents_path	Path to the file with short documents
size	Return an estimate of the dataset size.
size_duplicate_docs
size_short_docs
stem_dir_path

__init__(input_dir)

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(process_id, n_process, dup_sh, short_sh)

A generator producing all documents in the dataset.

name()

Human-readable name of tfhe dataset

short_documents_path()

Path to the file with short documents

size()

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

previous

modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.Dataset

next

modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaBooksDataset