update dataset and notebooks 1-2

statmlben · Jul 21, 2021 · a33d2e5 · a33d2e5
1 parent a84d923
commit a33d2e5
Show file tree

Hide file tree

Showing 5 changed files with 102,716 additions and 511,600 deletions.
diff --git a/dataset/pre-process.py b/dataset/pre-process.py
@@ -28,7 +28,7 @@
 le_user = preprocessing.LabelEncoder()
 df['user_id'] = le_user.fit_transform(df['user_id'])
 
-df = df[df['user_id'] < 10000]
+df = df[df['user_id'] < 2000]
 
 le_movie = preprocessing.LabelEncoder()
 df['movie_id'] = le_movie.fit_transform(df['movie_id'])