junte/ML
1
0
Fork 0
mirror of https://github.com/Hestia-Homes/ML.git synced 2026-06-08 11:17:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

trigger only one model to warm up, speed up dataframe generation

Browse source
This commit is contained in:
Michael Duong 2025-11-04 16:16:30 +00:00
parent 3b9699038a
commit 175bf87a95
1 changed files with 15 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
deployment/handlers/prediction_app.py
View file

@ -47,21 +47,28 @@ def upload_dataframe_to_s3(df, bucket, s3_file_name):
return False
def warming_up_invocation(model_filepath: str):
def warming_up_invocation(
model,
model_filepath: str,
):
"""
Function to handle warm up invocations
"""
import pandas as pd
import numpy as np
model = model_factory(settings.build_model["model_type"])
model_filepath = settings.build_model["model_save_filepath"]
model.load_model(model_filepath)
warmup_df = pd.DataFrame(columns=model.model.original_features)
warmup_df = pd.concat([warmup_df.T, pd.DataFrame([0] * len(warmup_df.T))], axis=1).T
warmup_df.fillna(0, inplace=True)
warmup_df = pd.DataFrame(
np.zeros((1, len(model.model.original_features))),
columns=model.model.original_features,
)
model.predict(data=warmup_df)
model_names = model.model.model_names()
if "NeuralNetFastAI" in model_names:
model.model.predict(warmup_df, model="NeuralNetFastAI")
else:
model.predict(data=warmup_df)
def handler(event, context):
@ -97,7 +104,7 @@ def handler(event, context):
if "warm" in body:
logger.info("Warm up invocation - synthetic prediction")
warming_up_invocation(model_filepath=model_filepath)
warming_up_invocation(model=model, model_filepath=model_filepath)
return {
"statusCode": 200,