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/dockerfile
+++ b/app/dockerfile
@@ -11,7 +11,7 @@ COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt

 # Copy the model file into the container
-COPY random_forest_model.pkl /app/
+COPY randomized_search_xgb_model-2.pkl /app/

 # Copy the rest of the working directory contents into the container at /app
 COPY . .