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

remove the specific fast ai prediction

Browse source
This commit is contained in:
Michael Duong 2025-11-04 16:52:24 +00:00
parent 9cd9208e3b
commit 3e38b572ce
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
deployment/handlers/prediction_app.py
View file

@ -64,11 +64,11 @@ def warming_up_invocation(
columns=model.model.original_features,
)
model_names = model.model.model_names()
if "NeuralNetFastAI" in model_names:
model.model.predict(warmup_df, model="NeuralNetFastAI")
else:
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):