Ekeany / Boruta-Shap

A Tree based feature selection tool which combines both the Boruta feature selection algorithm with shapley values.
MIT License
559 stars 86 forks source link

[ENH] Support for pairwise learning-to-rank models #62

Open poonia92 opened 3 years ago

poonia92 commented 3 years ago

Current Situation

Train_model function does not support training for pairwise leaning-to-rank models as these models requires a grouping of observations passed to their model.fit function.

An example of model.fit function for pairwise learning can be found here

Enhancement

Improve fit function by adding another function argument for grouping of observations.

Use this grouping of data in Train_model function and also in create_shadow_feature function while shuffling the columns.

Also as mentioned in issue-57, grouping of observations need to be taken into account for train-test split of data.

Reasoning

This will add support for all the pairwise learning-to-rank models.

Implementation

I'll be happy to share a PR in some time given the go-ahead