Skip to content

v0.6.0

Latest
Compare
Choose a tag to compare
@github-actions github-actions released this 29 Sep 22:33
de2b3c6

MLJFlux v0.6.0

Diff since v0.5.1

All models, except ImageClassifier, now support categorical features (presented as table columns with a CategoricalVector type). Rather than one-hot encoding, embeddings into a continuous space are learned (i.e, by adding an embedding layer) and the dimension of theses spaces can be specified by the user, using a new dictionary-valued hyperparameter, embedding_dims. The learned embeddings are exposed by a new implementation of transform, which means they can be used with other models (transfer learning) as described in Cheng Guo and Felix Berkhahn (2016): Entity Embeddings of Categorical Variables.

Also, all continuous input presented to these models is now forced to be Float32, but this is the only breaking change.

Merged pull requests:

Closed issues:

  • deprecated warning in documentation (#236)
  • Some minor doc issues (#258)
  • Fix code snippet in Readme (#266)
  • MultitargetNeuralNetworkRegressor doc example doesn't work as intended (#268)