Learn R Programming

R interface to TensorFlow Dataset API

The TensorFlow Dataset API provides various facilities for creating scalable input pipelines for TensorFlow models, including:

  • Reading data from a variety of formats including CSV files and TFRecords files (the standard binary format for TensorFlow training data).

  • Transforming datasets in a variety of ways including mapping arbitrary functions against them.

  • Shuffling, batching, and repeating datasets over a number of epochs.

  • Streaming interface to data for reading arbitrarily large datasets.

  • Reading and transforming data are TensorFlow graph operations, so are executed in C++ and in parallel with model training.

The R interface to TensorFlow datasets provides access to the Dataset API, including high-level convenience functions for easy integration with the keras package.

For documentation on using tfdatasets, see the package website at https://tensorflow.rstudio.com/tools/tfdatasets/.

Copy Link

Version

Install

install.packages('tfdatasets')

Monthly Downloads

2,396

Version

2.18.0

License

Apache License 2.0

Issues

15

Pull Requests

0

Stars

34

Forks

12

Maintainer

Tomasz Kalinowski

Last Published

August 20th, 2025

Functions in tfdatasets (2.18.0)

dataset_interleave

Maps map_func across this dataset, and interleaves the results
dataset_concatenate

Creates a dataset by concatenating given dataset with this dataset.
dataset_map

Map a function across a dataset.
dataset_map_and_batch

Fused implementation of dataset_map() and dataset_batch()
dataset_enumerate

Enumerates the elements of this dataset
dataset_decode_delim

Transform a dataset with delimted text lines into a dataset with named columns
dataset_filter

Filter a dataset by a predicate
dataset_flat_map

Maps map_func across this dataset and flattens the result.
dataset_group_by_window

Group windows of elements by key and reduce them
dataset_options

Get or Set Dataset Options
dataset_rebatch

Rebatch elements from this dataset into batches of specified size.
dataset_prefetch_to_device

A transformation that prefetches dataset values to the given device
dataset_rejection_resample

A transformation that resamples a dataset to a target distribution.
dataset_scan

A transformation that scans a function across an input dataset
dataset_reduce

Reduces the input dataset to a single element.
dataset_prefetch

Creates a Dataset that prefetches elements from this dataset.
dataset_repeat

Repeats a dataset count times.
dataset_padded_batch

Combines consecutive elements of this dataset into padded batches.
dataset_shard

Creates a dataset that includes only 1 / num_shards of this dataset.
dataset_prepare

Prepare a dataset for analysis
dataset_unique

A transformation that discards duplicate elements of a Dataset.
dataset_skip

Creates a dataset that skips count elements from this dataset
dataset_shuffle_and_repeat

Shuffles and repeats a dataset returning a new permutation for each epoch.
dataset_shuffle

Randomly shuffles the elements of this dataset.
dataset_snapshot

Persist the output of a dataset
dataset_take_while

A transformation that stops dataset iteration based on a predicate.
dataset_use_spec

Transform the dataset using the provided spec.
dataset_take

Creates a dataset with at most count elements from this dataset
dataset_window

Combines input elements into a dataset of windows.
dataset_unbatch

Unbatch a dataset
has_type

Identify the type of the variable.
iterator_get_next

Get next element from iterator
fit.FeatureSpec

Fits a feature specification.
hearts

Heart Disease Data Set
dense_features

Dense Features
fixed_length_record_dataset

A dataset of fixed-length records from one or more binary files.
file_list_dataset

A dataset of all files matching a pattern
delim_record_spec

Specification for reading a record from a text file with delimited values
iterator_initializer

An operation that should be run to initialize this iterator.
feature_spec

Creates a feature specification.
next_batch

Tensor(s) for retrieving the next batch from a dataset
layer_input_from_dataset

Creates a list of inputs from a dataset
random_integer_dataset

Creates a Dataset of pseudorandom values
output_types

Output types and shapes
length.tf_dataset

Get Dataset length
make-iterator

Creates an iterator for enumerating the elements of this dataset.
%>%

Pipe operator
iterator_string_handle

String-valued tensor that represents this iterator
iterator_make_initializer

Create an operation that can be run to initialize this iterator
make_csv_dataset

Reads CSV files into a batched dataset
scaler_standard

Creates an instance of a standard scaler
scaler_min_max

Creates an instance of a min max scaler
scaler

List of pre-made scalers
sql_record_spec

A dataset consisting of the results from a SQL query
reexports

Objects exported from other packages
read_files

Read a dataset from a set of files
sparse_tensor_slices_dataset

Splits each rank-N tf$SparseTensor in this dataset row-wise.
sample_from_datasets

Samples elements at random from the datasets in datasets.
range_dataset

Creates a dataset of a step-separated range of values.
selectors

Selectors
step_embedding_column

Creates embeddings columns
step_categorical_column_with_vocabulary_list

Creates a categorical column specification
step_crossed_column

Creates crosses of categorical columns
step_bucketized_column

Creates bucketized columns
step_categorical_column_with_identity

Create a categorical column with identity
step_numeric_column

Creates a numeric column specification
step_categorical_column_with_hash_bucket

Creates a categorical column with hash buckets specification
step_remove_column

Creates a step that can remove columns
step_categorical_column_with_vocabulary_file

Creates a categorical column with vocabulary file
step_indicator_column

Creates Indicator Columns
tensor_slices_dataset

Creates a dataset whose elements are slices of the given tensors.
text_line_dataset

A dataset comprising lines from one or more text files.
zip_datasets

Creates a dataset by zipping together the given datasets.
tfrecord_dataset

A dataset comprising records from one or more TFRecord files.
until_out_of_range

Execute code that traverses a dataset until an out of range condition occurs
tensors_dataset

Creates a dataset with a single element, comprising the given tensors.
with_dataset

Execute code that traverses a dataset
steps

Steps for feature columns specification.
step_shared_embeddings_column

Creates shared embeddings for categorical columns
all_nominal

Find all nominal variables.
dataset_bucket_by_sequence_length

A transformation that buckets elements in a Dataset by length
dataset_batch

Combines consecutive elements of this dataset into batches.
choose_from_datasets

Creates a dataset that deterministically chooses elements from datasets.
all_numeric

Speciy all numeric variables.
dataset_cache

Caches the elements in this dataset.
as_tf_dataset

Add the tf_dataset class to a dataset
as_array_iterator

Convert tf_dataset to an iterator that yields R arrays.
as_tensor.tensorflow.python.data.ops.dataset_ops.DatasetV2

Get the single element of the dataset.
dataset_collect

Collects a dataset