Open alexkaf opened 1 month ago
Hey thanks for raising the issue. Check out this notebook to see if it helps. Otherwise I'll look at it over the weekend because I haven't looked at Mockingjay in a while.
https://gist.github.com/acalejos/4598e5e2b2b91e420a4cf609bc2ffc03#compiling-the-exgboost-model
Same thing. All in all, the missing EXGBoost.compile/2
is not a problem since MockingJay.convert/2
can be used, the problem I see (and correct me if I'm wrong) is that the implementation of the MockingJay.DecisionTree
is missing for EXGBoost.Booster
.
Did you make sure to pull from the feature branch?
{:mockingjay,
git: "https://github.com/acalejos/mockingjay.git", branch: "make_tree_travs_jit_compilable"}
I'll take some time to try and clean it up and merge into main but it's been very low priority hence why its still in that state
I was following the Livebook provided in the docs. I noticed that there exists an
EXGBoost.compile/2
invocation, however this function does not exist in the current version and the notebook fails.Also, I searched older commits and noticed that
EXGBoost.compile/2
was essentially a simple invocaon toMockingjay.convert/2
function. Yet, even after replacing theEXGBoost.compile/1
withMockingjay.convert/2
on the notebook example, I get the following error:What is the correct approach of transforming an XGBoost loaded model to a series of tensor operations?