Skip to contents

Trains a LightGBM model on the complete dataset using the winning hyperparameters, serializes it to a text file, and uploads it directly to MinIO via the Apache Arrow S3 interface.

Usage

train_production_model(
  data,
  recipe,
  best_params,
  model_filename = "lgbm_prod.txt"
)

Arguments

data

A data frame containing the full BAF dataset (Months 0-7).

recipe

A prepared tidymodels recipe.

best_params

A list or tibble of the winning hyperparameters.

model_filename

Character. The target filename. Defaults to "lgbm_prod.txt".

Value

Character. The MinIO URI of the uploaded model artifact.