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

add time to inference to model

Browse source
This commit is contained in:
Michael Duong 2023-10-22 21:05:07 +00:00
parent 499458b699
commit 0f96bc55f1
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
modules/ml-pipeline/src/pipeline/core/MLModels.py
View file

@ -149,6 +149,8 @@ class AutogluonAutoML:
"time_limit",
"presets",
"excluded_model_types",
"infer_limit",
"infer_limit_batch_size",
]
def load_model(self, path: Union[Path, str]) -> None:
@ -203,6 +205,8 @@ class AutogluonAutoML:
time_limit=model_hyperparameters["time_limit"],
presets=model_hyperparameters["presets"],
excluded_model_types=model_hyperparameters["excluded_model_types"],
infer_limit=model_hyperparameters["infer_limit"],
infer_limit_batch_size=model_hyperparameters["infer_limit_batch_size"],
)
def predict(