import tensorflow as tf

features = {
    "sepal_length": tf.constant([5.1, 4.9, 6.7, 5.6], dtype=tf.float32),
    "sepal_width": tf.constant([3.5, 3.0, 3.1, 2.8], dtype=tf.float32),
}
labels = tf.constant([0, 0, 2, 1], dtype=tf.int32)