import tensorflow as tf



_TRAIN_FEATURE_MAP = {
    movielens.USER_COLUMN: tf.FixedLenFeature([], dtype=tf.string),
    movielens.ITEM_COLUMN: tf.FixedLenFeature([], dtype=tf.string),
    rconst.MASK_START_INDEX: tf.FixedLenFeature([1], dtype=tf.string),
    "labels": tf.FixedLenFeature([], dtype=tf.string),
}


