Skip to content

unbundle issue with XGBoost models #81

@lanedrew

Description

@lanedrew

When using the unbundle function on a bundled XGBoost model, even after the 0.1.3 update, there is an error associated with the argument as_booster which has become deprecated.

Below is an example of the output received when attempting to unbundle a model after loading the bundled model object:

Load the example mark model that previously was trained on the small example data

file_path <- system.file("extdata", "example_mark_model.rds", package = "ldmppr")
mark_model <- bundle::unbundle(readRDS(file_path))
Error in xgboost::xgb.load.raw(object, as_booster = TRUE) :
unused argument (as_booster = TRUE)
Calls: ... -> unbundle -> unbundle.bundle ->
Execution halted

From what I can tell, just removing the as_booster argument should address the issue in the unbundle function for XGBoost objects.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions