nandovallec commited on
Commit
f8c1a4b
1 Parent(s): 9cb5f62

Optimization

Browse files
Files changed (2) hide show
  1. app.py +6 -0
  2. recommender.py +1 -9
app.py CHANGED
@@ -43,6 +43,12 @@ repo_mat = Repository(
43
  local_dir="data_mat", clone_from=DATASET_REPO_URL_MAT, use_auth_token=HF_TOKEN, repo_type="dataset"
44
  )
45
 
 
 
 
 
 
 
46
  from fetchPlaylistTrackUris import *
47
  from recommender import *
48
 
 
43
  local_dir="data_mat", clone_from=DATASET_REPO_URL_MAT, use_auth_token=HF_TOKEN, repo_type="dataset"
44
  )
45
 
46
+ df_ps_train_ori = pd.read_hdf('model/df_ps_train_new.hdf')
47
+ df_ps_train_extra = pd.read_hdf('data_train/df_ps_train_extra.hdf')
48
+ pickle_path = 'model/giantMatrix_new.pickle'
49
+ with open(pickle_path, 'rb') as f:
50
+ ps_matrix_ori = pickle.load(f)
51
+
52
  from fetchPlaylistTrackUris import *
53
  from recommender import *
54
 
recommender.py CHANGED
@@ -4,15 +4,7 @@ from scipy.sparse import csr_matrix
4
  import numpy as np
5
  import pandas as pd
6
  from scipy.sparse import vstack
7
-
8
-
9
- df_ps_train_ori = pd.read_hdf('model/df_ps_train_new.hdf')
10
- df_ps_train_extra = pd.read_hdf('data_train/df_ps_train_extra.hdf')
11
- pickle_path = 'model/giantMatrix_new.pickle'
12
- with open(pickle_path, 'rb') as f:
13
- ps_matrix_ori = pickle.load(f)
14
-
15
-
16
 
17
  def add_row_train(df, list_tid):
18
  new_pid_add = df.iloc[-1].name +1
 
4
  import numpy as np
5
  import pandas as pd
6
  from scipy.sparse import vstack
7
+ from app import
 
 
 
 
 
 
 
 
8
 
9
  def add_row_train(df, list_tid):
10
  new_pid_add = df.iloc[-1].name +1