Spaces:
Sleeping
Sleeping
Update src/recommendation_utils.py
Browse files
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 |
-
|
|
|
|
| 22 |
with open(path, "rb") as f:
|
| 23 |
return pickle.load(f)
|
| 24 |
|
| 25 |
-
|
|
|
|
| 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 |
# βββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββββ
|