"examples/mxnet/vscode:/vscode.git/clone" did not exist on "97e792654ff535140495ccb2e183f8e0117998b7"
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):
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)
......
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