Skip to content

0.1.8

Choose a tag to compare

@achoum achoum released this 29 Jul 14:55
· 582 commits to main since this release

Features

  • Model can be composed with the functional Keras API before being trained.
  • Makes all the Yggdrasil structural variable importances available.
  • Makes getting the variable importance instantaneous.
  • Surface the name argument in the model classes constructors.
  • Add a postprocessing model constructor argument to easy apply
    post-processing on the model predictions without relying on the Keras
    Functional API.
  • Add extract_all_trees method in the model inspector to efficiently exact
    all the trees.
  • Add num_threads constructor argument to control the number of training
    threads without using the advanced configuration.
  • By default, remove the temporary directory used to train the model when the
    model python object is garbage collected.
  • Add the import_dataspec constructor argument to the model builder to
    import the feature definition and dictionaries (instead of relying on
    automatic discovery).

Changes

  • When saving a model in a directory already containing a model, only the
    assets directory is entirely removed before the export (instead of the
    entire model directory).

Fixes

  • Wrong label shape in the model inspector's objective field for
    pre-integerized labels.