Polo123 commited on
Commit
82265a5
·
verified ·
1 Parent(s): 0c181d0

Update logic.py

Browse files
Files changed (1) hide show
  1. logic.py +1 -0
logic.py CHANGED
@@ -220,6 +220,7 @@ class Model(torch.nn.Module):
220
 
221
  #-------------------------------------------------------------------------------------------
222
  def make_graph():
 
223
  metadata_path = './sampled_movie_dataset/movies_metadata.csv'
224
  df = pd.read_csv(metadata_path)
225
  df = df.drop([19730, 29503, 35587])
 
220
 
221
  #-------------------------------------------------------------------------------------------
222
  def make_graph():
223
+ global movie_mappings, user_mapping, ratings_df, m_id, id_map, sampled_md
224
  metadata_path = './sampled_movie_dataset/movies_metadata.csv'
225
  df = pd.read_csv(metadata_path)
226
  df = df.drop([19730, 29503, 35587])