diff --git a/Dockerfile b/Dockerfile index 1fdb49d..476d6d2 100644 --- a/Dockerfile +++ b/Dockerfile @@ -30,7 +30,7 @@ RUN install2.r --error --skipinstalled \ RUN Rscript -e "remotes::install_version('h2o', '3.42.0.2')" # Add the prediction model to the app -COPY IIT-Prediction/model/V9 /app/model +COPY IIT-Prediction/model/V9B /app/model # Add the production extraction query to the app COPY SQL/iit_prod_data_extract.sql /app/iit_prod_data_extract.sql diff --git a/docker-resources/plumber.R b/docker-resources/plumber.R index e073579..1e736f4 100644 --- a/docker-resources/plumber.R +++ b/docker-resources/plumber.R @@ -24,12 +24,12 @@ ml_model_version <- "V9" # now we load the models into the h2o server instance # adult model ml_model_adult <- h2o.loadModel( - "/app/model/y0_1days_adult_IIT/2_StackedEnsemble_BestOfFamily_1_AutoML_8_20240411_135528_auc_0.739/StackedEnsemble_BestOfFamily_1_AutoML_8_20240411_135528" + "/app/model/y0_1days_adult_IIT/2_StackedEnsemble_BestOfFamily_1_AutoML_1_20240513_202609_auc_0.783/StackedEnsemble_BestOfFamily_1_AutoML_1_20240513_202609" ) # peds model ml_model_minor <- h2o.loadModel( - "/app/model/y0_1days_minor_IIT/1_StackedEnsemble_AllModels_1_AutoML_6_20240329_151542_auc_0.721/StackedEnsemble_AllModels_1_AutoML_6_20240329_151542" + "/app/model/y0_1days_minor_IIT/2_StackedEnsemble_BestOfFamily_1_AutoML_2_20240513_235011_auc_0.725/StackedEnsemble_BestOfFamily_1_AutoML_2_20240513_235011" ) # we also load the SQL query we use to generate the dataframe of records for prediction