Skip to content

Commit 013bc03

Browse files
committed
fix error in run_ncf.py
1 parent 93ca437 commit 013bc03

1 file changed

Lines changed: 1 addition & 3 deletions

File tree

examples/run_ncf.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,14 +5,12 @@
55

66
if __name__ == "__main__":
77
data = pd.read_csvdata = pd.read_csv("./movielens_sample.txt")
8-
sparse_features = ["movie_id", "user_id",
9-
"gender", "age", "occupation", "zip", ]
108
SEQ_LEN = 50
119
negsample = 3
1210

1311
# 1.Label Encoding for sparse features,and process sequence features with `gen_date_set` and `gen_model_input`
1412

15-
features = ['user_id', 'movie_id', 'gender', 'age', 'occupation', 'zip']
13+
features = ['user_id', 'movie_id', 'gender', 'age', 'occupation', 'zip', "genres"]
1614
feature_max_idx = {}
1715
for feature in features:
1816
lbe = LabelEncoder()

0 commit comments

Comments
 (0)