Skip to content

Save model to json or universal binary json rather than pkl #392

@kavidey

Description

@kavidey

For large models, saving models using pickle can create large files (1 GB+) and can create compatibility issues that require specific library versions. It would be awesome if there was a more portable, archivable format that models could be saved/loaded in such as json or ubj. Additionally if space could be saved by only saving the parts of the model needed for inference (eg not feature importance) that would be great.

I'm happy to help with the implementation, but am not as familiar with the ngboost code and which parts of the model would need to be serialized.

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