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

use model columns as data

Browse source
This commit is contained in:
Michael Duong 2025-11-03 22:40:35 +00:00
parent 2c8f014258
commit d7370248a2
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
deployment/handlers/prediction_app.py
View file

@ -90,7 +90,9 @@ def handler(event, context):
return {
"statusCode": 200,
"body": json.dumps(
{"message": f"{model.predict(data=pd.DataFrame({'a': [1]}))}"}
{
"message": f"{model.predict(data=pd.DataFrame(columns=model.model.original_features))}"
}
),
}
else: