Fix docstrings in data_preprocessing.py. (tensorflow#4976)

reedwm · Taylor Robie · commit f0e10716160c · 2018-08-02T12:52:19.000-07:00
diff --git a/official/recommendation/data_preprocessing.py b/official/recommendation/data_preprocessing.py
@@ -59,6 +59,8 @@ def __init__(self, user_map, item_map, num_data_readers, cache_paths,
       item_map: Dict mapping raw item ids to regularized ids.
       num_data_readers: The number of reader Datasets used during training.
       cache_paths: Object containing locations for various cache files.
+      num_train_positives: The number of positive training examples in the
+        dataset.
     """
 
     self.user_map = {int(k): int(v) for k, v in user_map.items()}
@@ -255,9 +257,6 @@ def generate_train_eval_data(df, approx_num_shards, num_items, cache_paths):
     num_items: The cardinality of the item set.
     cache_paths: rconst.Paths object containing locations for various cache
       files.
-
-  Returns:
-    A tuple containing the validation data.
   """
 
   num_rows = len(df)
