data.rst 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758
  1. .. role:: hidden
  2. :class: hidden-section
  3. .. module:: fairseq.data
  4. Data Loading and Utilities
  5. ==========================
  6. .. _datasets:
  7. Datasets
  8. --------
  9. **Datasets** define the data format and provide helpers for creating
  10. mini-batches.
  11. .. autoclass:: fairseq.data.FairseqDataset
  12. :members:
  13. .. autoclass:: fairseq.data.LanguagePairDataset
  14. :members:
  15. .. autoclass:: fairseq.data.MonolingualDataset
  16. :members:
  17. **Helper Datasets**
  18. These datasets wrap other :class:`fairseq.data.FairseqDataset` instances and
  19. provide additional functionality:
  20. .. autoclass:: fairseq.data.BacktranslationDataset
  21. :members:
  22. .. autoclass:: fairseq.data.ConcatDataset
  23. :members:
  24. .. autoclass:: fairseq.data.ResamplingDataset
  25. :members:
  26. .. autoclass:: fairseq.data.RoundRobinZipDatasets
  27. :members:
  28. .. autoclass:: fairseq.data.TransformEosDataset
  29. :members:
  30. Dictionary
  31. ----------
  32. .. autoclass:: fairseq.data.Dictionary
  33. :members:
  34. Iterators
  35. ---------
  36. .. autoclass:: fairseq.data.CountingIterator
  37. :members:
  38. .. autoclass:: fairseq.data.EpochBatchIterator
  39. :members:
  40. .. autoclass:: fairseq.data.GroupedIterator
  41. :members:
  42. .. autoclass:: fairseq.data.ShardedIterator
  43. :members: