change to xgb classifier

2024-08-14 01:01:00 +08:00
parent bd41e02b26
commit 1e081c7e47
5 changed files with 3 additions and 2 deletions
--- a/app/main.py
+++ b/app/main.py
@@ -7,7 +7,7 @@ import pandas as pd
 app = FastAPI()

 # Load the trained model
-model = joblib.load('random_forest_model.pkl')
+model = joblib.load('randomized_search_xgb_model-2.pkl')

 # Define the input data structure using Pydantic
 class InputData(BaseModel):