dkoshman commited on
Commit
c71d708
1 Parent(s): 3047c04

my_recommending==0.1.19, catboost aggregator

Browse files
Files changed (2) hide show
  1. app.py +6 -4
  2. requirements.txt +1 -1
app.py CHANGED
@@ -12,23 +12,25 @@ project = "Recommending"
12
  tmdb_api_token = os.environ["TMDB_API_TOKEN"]
13
  lightning_class = MovieLensNonGradientRecommender
14
  config = dict(
15
- model_artifact="my_mf_slim_movielens_25m:latest",
16
- data_artifact="movielens25m:latest",
17
  media_directory="media",
18
  n_recommendations_from_imdb_ratings=50,
19
  )
20
 
21
  wandb.init(job_type="app", project=project, config=config)
22
 
 
 
 
23
  model_artifact = wandb.use_artifact(config["model_artifact"])
24
  checkpoint_path = model_artifact.file()
 
25
  lightning_module = lightning_class.load_from_checkpoint(
26
  checkpoint_path, map_location="cpu"
27
  )
28
  recommender = lightning_module.model.eval()
29
 
30
- data_artifact = wandb.use_artifact(config["data_artifact"])
31
- data_directory = data_artifact.download()
32
  movielens = MovieLens25m(directory=data_directory)
33
  movie_markdown_generator = MovieMarkdownGenerator(
34
  movielens=movielens, tmdb_api_token=tmdb_api_token
 
12
  tmdb_api_token = os.environ["TMDB_API_TOKEN"]
13
  lightning_class = MovieLensNonGradientRecommender
14
  config = dict(
15
+ model_artifact="catboost_aggregator_25m:v5",
16
+ data_artifact="movielens25m:v4",
17
  media_directory="media",
18
  n_recommendations_from_imdb_ratings=50,
19
  )
20
 
21
  wandb.init(job_type="app", project=project, config=config)
22
 
23
+ data_artifact = wandb.use_artifact(config["data_artifact"])
24
+ data_directory = data_artifact.download()
25
+
26
  model_artifact = wandb.use_artifact(config["model_artifact"])
27
  checkpoint_path = model_artifact.file()
28
+
29
  lightning_module = lightning_class.load_from_checkpoint(
30
  checkpoint_path, map_location="cpu"
31
  )
32
  recommender = lightning_module.model.eval()
33
 
 
 
34
  movielens = MovieLens25m(directory=data_directory)
35
  movie_markdown_generator = MovieMarkdownGenerator(
36
  movielens=movielens, tmdb_api_token=tmdb_api_token
requirements.txt CHANGED
@@ -1 +1 @@
1
- my_recommending==0.1.18
 
1
+ my_recommending==0.1.19