jrno commited on
Commit
d4857a9
1 Parent(s): 63d0aa5

Update user history to use dataset model was trained on

Browse files
data/user_listening_history_10k.csv DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:47a82d52ec512f00bf1a3416ecbf153aaa478266e87f6d3c0c4bff85ce4e1d4a
3
- size 620427
 
 
 
 
recommendation-api/model.pkl CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:572c13e535c670418a3391e69edfeaa6249964a247bf9dcf978ca15333b9b494
3
- size 10347577
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:469699967d2f8722c960d94ca566bd4401837cc76a9b60de70fe1141a15a87f2
3
+ size 10578681
recommendation-api/tracks.py CHANGED
@@ -4,7 +4,7 @@ import pandas as pd
4
  tracks_df = pd.read_csv('data/music_info.csv')
5
  tracks_df.fillna('', inplace=True)
6
  tracks_df["entry"] = tracks_df["name"] + ", " + tracks_df["artist"] + ", " + tracks_df["year"].astype(str)
7
- track_interactions_df = pd.read_csv('data/user_listening_history_10k.csv')
8
 
9
  # Merge data on those two csvs
10
  dataframe = pd.merge(tracks_df, track_interactions_df, on='track_id', how='left')
 
4
  tracks_df = pd.read_csv('data/music_info.csv')
5
  tracks_df.fillna('', inplace=True)
6
  tracks_df["entry"] = tracks_df["name"] + ", " + tracks_df["artist"] + ", " + tracks_df["year"].astype(str)
7
+ track_interactions_df = pd.read_csv('data/model_track_interactions.csv')[['user_id', 'track_id']]
8
 
9
  # Merge data on those two csvs
10
  dataframe = pd.merge(tracks_df, track_interactions_df, on='track_id', how='left')