def make_sequence_example()

in recommended-item-search/data_preparation.py [0:0]

• 11 lines of code
• 2 McCabe index (conditional complexity)

def make_sequence_example(user_id, movie_ids):
  """Returns a SequenceExample for the given user_id and movie_ids.
  Args:
    user_id: An user_id.
    movie_ids: A list of strings.
  Returns:
    A tf.train.SequenceExample containing user_id and movie_ids.
  """
  def _int64_feature(value):
    return tf.train.Feature(int64_list=tf.train.Int64List(value=[value]))

  feature = {'user_id': _int64_feature(user_id)}
  movie_id_features = [
      tf.train.Feature(int64_list=tf.train.Int64List(value=[id_]))
      for id_ in movie_ids]
  feature_list = {'movie_ids': tf.train.FeatureList(feature=movie_id_features)}
  features = tf.train.Features(feature=feature)
  feature_lists = tf.train.FeatureLists(feature_list=feature_list)
  return tf.train.SequenceExample(context=features, feature_lists=feature_lists)