search

dmlc / tl2cgen

TL2cgen (TreeLite 2 C GENerator) is a model compiler for decision tree models
https://tl2cgen.readthedocs.io/en/latest/
Apache License 2.0
17 stars 6 forks source link

[FEA]Add max_memory_usage param in model.export_lib #7

Open ysq151944 opened 1 year ago

ysq151944 commented 1 year ago

Treelite need above 600+GB memory to compile a large sklearn RF model(106MB after joblib compress).

So it would be great if I can set a maximum memory usage when compling the model