Repository navigation
v1.3.0
NeuroTreeModels v1.3.0
Important changes
- The kwarg
device(:cpu/:gpu) is moved fromNeuroTreeRegressortofit. Same forgpuID. - Removal of
outsizeargument. - Introduction of
NeuroTreeClassififier, respecting the MLJ interface - It's no longer needed to specify the number of classes (through the deprecated
outsizekwarg). They're now automatically detected throuh the target variable number of levels. - Classification tasks (using
NeuroTreeClassififier) require the target variable to be<:Categorical
Merged pull requests:
- add tweedie (#11) (@jeremiedb)
- Dev (#12) (@jeremiedb)
Closed issues:
- Robust preprocessing of target for classifiction tasks (#10)