Input function for estimator in TensorFlow

tf_l1_reg_input_fn.py

def input_fn(X, y, batch_size=16, epochs=1, shuffle=False):
  #create a dictionary of features and their values
  features = {key:np.array(value) for key,value in dict(X).items()} 
  #create a tf.data compliant dataset
  d = tf.data.Dataset.from_tensor_slices((features, y))
  
  #we need our data in batches
  d = d.batch(batch_size).repeat(epochs)
  
  #optionally shuffle our data
  if shuffle:
    d.shuffle(10000)
  #make an iterator
  iterator = d.make_one_shot_iterator()
  return iterator.get_next()