Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "9bcbe962e95ea99582db3aec10225f9d03bfa224"
Commit f0e10716 authored by Reed's avatar Reed Committed by Taylor Robie
Browse files

Fix docstrings in data_preprocessing.py. (#4976)

parent 58037d2c
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
official/recommendation/data_preprocessing.py
View file @ f0e10716
...@@ -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