Spaces:
Sleeping
Sleeping
Update src/recommendation_utils.py
Browse files- src/recommendation_utils.py +12 -0
src/recommendation_utils.py
CHANGED
|
@@ -2,6 +2,18 @@ import numpy as np
|
|
| 2 |
import pandas as pd
|
| 3 |
import pickle
|
| 4 |
from keras.models import load_model
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
|
| 6 |
# Lade Modelle und Daten
|
| 7 |
def load_nn_model(path):
|
|
|
|
| 2 |
import pandas as pd
|
| 3 |
import pickle
|
| 4 |
from keras.models import load_model
|
| 5 |
+
from keras.models import model_from_json
|
| 6 |
+
from keras.optimizers import Adam
|
| 7 |
+
|
| 8 |
+
def load_nn_model(config_path, weights_path):
|
| 9 |
+
with open(config_path, "r") as f:
|
| 10 |
+
model_json = f.read()
|
| 11 |
+
model = model_from_json(model_json)
|
| 12 |
+
model.load_weights(weights_path)
|
| 13 |
+
|
| 14 |
+
# Gleiche Konfiguration wie im Training:
|
| 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):
|