Commit f0e10716 authored by Reed's avatar Reed Committed by Taylor Robie
Browse files

Fix docstrings in data_preprocessing.py. (#4976)

parent 58037d2c
...@@ -59,6 +59,8 @@ class NCFDataset(object): ...@@ -59,6 +59,8 @@ class NCFDataset(object):
item_map: Dict mapping raw item ids to regularized ids. item_map: Dict mapping raw item ids to regularized ids.
num_data_readers: The number of reader Datasets used during training. num_data_readers: The number of reader Datasets used during training.
cache_paths: Object containing locations for various cache files. 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()} 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): ...@@ -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. num_items: The cardinality of the item set.
cache_paths: rconst.Paths object containing locations for various cache cache_paths: rconst.Paths object containing locations for various cache
files. files.
Returns:
A tuple containing the validation data.
""" """
num_rows = len(df) num_rows = len(df)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment