Skip to content

v1.3.0

Choose a tag to compare

@github-actions github-actions released this 21 Apr 22:52

NeuroTreeModels v1.3.0

Important changes

  • The kwarg device (:cpu / :gpu) is moved from NeuroTreeRegressor to fit. Same for gpuID.
  • Removal of outsize argument.
  • Introduction of NeuroTreeClassififier, respecting the MLJ interface
  • It's no longer needed to specify the number of classes (through the deprecated outsize kwarg). They're now automatically detected throuh the target variable number of levels.
  • Classification tasks (using NeuroTreeClassififier) require the target variable to be <:Categorical

Diff since v1.2.0

Merged pull requests:

Closed issues:

  • Robust preprocessing of target for classifiction tasks (#10)