tomtung / omikuji

An efficient implementation of Partitioned Label Trees & its variations for extreme multi-label classification

Home Page:https://crates.io/crates/omikuji

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Cannot set collapse_every_n_layers via Python bindings

osma opened this issue · comments

Hi,

I'm trying to figure out how to use the AttentionXML-like hyperparameters mentioned in the top level README using the Python bindings. This would require setting the collapse_every_n_layers hyperparameter to 5.

It appears to me that this hyperparameter is not exposed to the Python bindings (nor to the C API). There is no field called collapse_every_n_layers (or anything similar) in the HyperParam object returned by omikuji.Model.default_hyper_param(). In fact, the only source files that mention this hyperparameter are src/bin/omikuji.rs and src/model/train.rs so it is only used within the Rust implementation, not in any of the bindings.

Should be fixed in the newest release.

Thanks, this was fast! It works very well now in version 0.2.0.