Spaces:
Sleeping
Sleeping
Update src/recommendation_utils.py
Browse files
src/recommendation_utils.py
CHANGED
|
@@ -11,14 +11,10 @@ def load_nn_model(config_path, weights_path):
|
|
| 11 |
model = model_from_json(model_json)
|
| 12 |
model.load_weights(weights_path)
|
| 13 |
|
| 14 |
-
#
|
| 15 |
model.compile(optimizer=Adam(), loss="mse", metrics=["mse", "mae"])
|
| 16 |
return model
|
| 17 |
|
| 18 |
-
# Lade Modelle und Daten
|
| 19 |
-
def load_nn_model(path):
|
| 20 |
-
return load_model(path)
|
| 21 |
-
|
| 22 |
def load_svd_model(path):
|
| 23 |
with open(path, "rb") as f:
|
| 24 |
return pickle.load(f)
|
|
|
|
| 11 |
model = model_from_json(model_json)
|
| 12 |
model.load_weights(weights_path)
|
| 13 |
|
| 14 |
+
# same config as used in training
|
| 15 |
model.compile(optimizer=Adam(), loss="mse", metrics=["mse", "mae"])
|
| 16 |
return model
|
| 17 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
def load_svd_model(path):
|
| 19 |
with open(path, "rb") as f:
|
| 20 |
return pickle.load(f)
|