JuliaAI / MLJXGBoostInterface.jl

MIT License
11 stars 5 forks source link

Overload MLJModelInterface.fitted_params to get more readable `fitted_params(mach)` output #35

Open ablaom opened 1 year ago

ablaom commented 1 year ago

Here's how things look for the classifier:

julia> fitted_params(mach)
(fitresult = (Booster(), CategoricalArrays.CategoricalValue{String, UInt32} "setosa"),)

Rather we should return something like (core_booster=..., encoding=...). The format of encoding could match what DecisionTreeClassifier does.

The other two models just have a core_booster.