Commit 97254e2d authored by Yomal-Dulanjana's avatar Yomal-Dulanjana

ML Update

parent 0f94abec
...@@ -47,7 +47,7 @@ diet_plan_mapping = { ...@@ -47,7 +47,7 @@ diet_plan_mapping = {
##################################### LOAD MODELS ######################################### ##################################### LOAD MODELS #########################################
risk_model = tf.keras.models.load_model('weights/risk_prediction.h5') risk_model = tf.keras.models.load_model('weights/future_prediction.h5')
risk_model.compile( risk_model.compile(
optimizer='adam', optimizer='adam',
loss='categorical_crossentropy', loss='categorical_crossentropy',
...@@ -58,7 +58,7 @@ risk_model.compile( ...@@ -58,7 +58,7 @@ risk_model.compile(
tf.keras.metrics.AUC() tf.keras.metrics.AUC()
]) ])
with open('weights/scalar - risk_prediction.pkl', 'rb') as f: with open('weights/scalar_future_prediction.pkl', 'rb') as f:
risk_scalar = pickle.load(f) risk_scalar = pickle.load(f)
...@@ -70,7 +70,7 @@ diet_workout_model.compile( ...@@ -70,7 +70,7 @@ diet_workout_model.compile(
metrics=['accuracy'] metrics=['accuracy']
) )
with open('weights/scalar - diet_workout_prediction.pkl', 'rb') as f: with open('weights/scalar_diet_workout_prediction.pkl', 'rb') as f:
diet_workout_scalar = pickle.load(f) diet_workout_scalar = pickle.load(f)
##################################### PREDICTION ######################################### ##################################### PREDICTION #########################################
......
This source diff could not be displayed because it is too large. You can view the blob instead.
This diff is collapsed.
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment