Как следует из названия, он кэширует образцы в памяти (или на диске, если указано имя файла). Например, если train_examples включает дорогостоящую предварительную обработку, предварительную обработку не нужно повторять для каждой эпохи за счет увеличения потребления памяти.
Обратите внимание, что из-за порядка все методы вызываются после .cache() по-прежнему выполняются каждую эпоху. В зависимости от ситуации может быть выгодно позвонить по номеру .map() перед .cache(). Однако перетасовка и пакетирование обычно должны вызываться в последнюю очередь.
Подробности см. В официальной документации .