How can Tensorflow be used to create an input function to to train the model?

An input function that would be used to train or evaluate the model can be created in Tensorflow by using the ‘from_tensor_slices’ method and creating a dictionary of the features of the iris dataset.

Read More: What is TensorFlow and how Keras work with TensorFlow to create Neural Networks?

We will use the Keras Sequential API, which is helpful in building a sequential model that is used to work with a plain stack of layers, where every layer has exactly one input tensor and one output tensor.

A neural network that contains at least one layer is known as a convolutional layer. We can use the Convolutional Neural Network to build learning model. 

TensorFlow Text contains collection of text related classes and ops that can be used with TensorFlow 2.0. The TensorFlow Text can be used to preprocess sequence modelling.

We are using the Google Colaboratory to run the below code. Google Colab or Colaboratory helps run Python code over the browser and requires zero configuration and free access to GPUs (Graphical Processing Units). Colaboratory has been built on top of Jupyter Notebook.


print("Input function that is used to train or evaluate")
def input_fn(features, labels, training=True, batch_size=256):
   dataset =, labels))
   if training:
      dataset = dataset.shuffle(1000).repeat()
   return dataset.batch(batch_size)


Input function that is used to train or evaluate

Code credit −


  • The input function may generate the features dictionary and label list.

  • Data is loaded using pandas, and an input pipeline is built from this in-memory data

Updated on: 22-Feb-2021


Kickstart Your Career

Get certified by completing the course

Get Started