lenawilli commited on
Commit
c0bc9f5
Β·
verified Β·
1 Parent(s): b84aad8

Update src/recommendation_utils.py

Browse files
Files changed (1) hide show
  1. src/recommendation_utils.py +8 -4
src/recommendation_utils.py CHANGED
@@ -1,8 +1,9 @@
 
1
  import numpy as np
2
  import pandas as pd
3
  import pickle
4
- from keras.models import model_from_json
5
- from keras.optimizers import Adam
6
 
7
  # ─────────────────────────────────────────────────────────────────────────────
8
  # Neural Network Model Laden (aus JSON + Weights)
@@ -18,14 +19,17 @@ def load_nn_model(config_path, weights_path):
18
  # ─────────────────────────────────────────────────────────────────────────────
19
  # SVD Model & Trainset Laden
20
  # ─────────────────────────────────────────────────────────────────────────────
21
- def load_svd_model(path):
 
22
  with open(path, "rb") as f:
23
  return pickle.load(f)
24
 
25
- def load_trainset(path):
 
26
  with open(path, "rb") as f:
27
  return pickle.load(f)
28
 
 
29
  # ─────────────────────────────────────────────────────────────────────────────
30
  # Encodings (dict mit user/movie Encodings) laden
31
  # ─────────────────────────────────────────────────────────────────────────────
 
1
+ import streamlit as st
2
  import numpy as np
3
  import pandas as pd
4
  import pickle
5
+ #from keras.models import model_from_json
6
+ #from keras.optimizers import Adam
7
 
8
  # ─────────────────────────────────────────────────────────────────────────────
9
  # Neural Network Model Laden (aus JSON + Weights)
 
19
  # ─────────────────────────────────────────────────────────────────────────────
20
  # SVD Model & Trainset Laden
21
  # ─────────────────────────────────────────────────────────────────────────────
22
+ @st.cache_resource
23
+ def load_svd_model(path="models/svd_model.pkl"):
24
  with open(path, "rb") as f:
25
  return pickle.load(f)
26
 
27
+ @st.cache_resource
28
+ def load_trainset(path="models/trainset.pkl"):
29
  with open(path, "rb") as f:
30
  return pickle.load(f)
31
 
32
+
33
  # ─────────────────────────────────────────────────────────────────────────────
34
  # Encodings (dict mit user/movie Encodings) laden
35
  # ─────────────────────────────────────────────────────────────────────────────