NVIDIA / spark-rapids-tools

User tools for Spark RAPIDS
Apache License 2.0
49 stars 36 forks source link

[FEA] Support custom XGBoost model file via user tools CLI #1134

Closed mattahrens closed 2 months ago

mattahrens commented 3 months ago

As a customer, I may wish to provide my own XGBoost estimation model file to use in qualification given that I've trained it on my own event logs. The user tools CLI should support a --custom_model_file argument (or something like that) and then use that for estimation during qualification.