jrno commited on
Commit
0d7a9d2
β€’
1 Parent(s): cfbd309

revert to old model.pkl

Browse files
Dockerfile CHANGED
@@ -6,8 +6,8 @@ COPY recommendation-api .
6
  COPY data/music_info.csv data/music_info.csv
7
  COPY data/model.csv data/model.csv
8
  COPY data/model.pkl data/model.pkl
9
- COPY data/model_25k.csv data/model_25k.csv
10
- COPY data/model_25k.pkl data/model_25k.pkl
11
 
12
  # Install dependencies from the requirements.txt file located within the recommendation-api directory
13
  RUN pip install -r requirements.txt
 
6
  COPY data/music_info.csv data/music_info.csv
7
  COPY data/model.csv data/model.csv
8
  COPY data/model.pkl data/model.pkl
9
+ COPY data/model_next_25k.csv data/model_next_25k.csv
10
+ COPY data/model_next_25k.pkl data/model_next_25k.pkl
11
 
12
  # Install dependencies from the requirements.txt file located within the recommendation-api directory
13
  RUN pip install -r requirements.txt
data/{model_25k.csv β†’ model_next_25k.csv} RENAMED
File without changes
data/{model_25k.pkl β†’ model_next_25k.pkl} RENAMED
File without changes
recommendation-api/server.py CHANGED
@@ -10,7 +10,7 @@ from recommender import get_recommendations_for_user
10
  from learner import setup_learner, custom_accuracy
11
 
12
  app = FastAPI()
13
- model_filename = 'data/model_25k.pkl'
14
  learn = None
15
 
16
  @app.on_event("startup")
 
10
  from learner import setup_learner, custom_accuracy
11
 
12
  app = FastAPI()
13
+ model_filename = 'data/model.pkl'
14
  learn = None
15
 
16
  @app.on_event("startup")
recommendation-api/tracks.py CHANGED
@@ -10,7 +10,7 @@ tracks_df["entry"] = tracks_df["name"] + ", " + tracks_df["artist"] + ", " + tra
10
  print("Music info parsed")
11
 
12
  # Raw dataframe from the training set
13
- model_df = pd.read_csv('data/model_25k.csv')
14
  model_interactions_df = model_df[['user_id', 'track_id']]
15
  model_tracks_df = model_df[['entry']].drop_duplicates()
16
 
 
10
  print("Music info parsed")
11
 
12
  # Raw dataframe from the training set
13
+ model_df = pd.read_csv('data/model.csv')
14
  model_interactions_df = model_df[['user_id', 'track_id']]
15
  model_tracks_df = model_df[['entry']].drop_duplicates()
16