Skip to content

My quick take on matching algorithms using RF, XGBoost on an imbalanced dataset.

Notifications You must be signed in to change notification settings

franciscogmm/MatchingAlgorithms

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

MatchingAlgorithms

My quick take on matching algorithms using RF, XGBoost on an imbalanced dataset.

The data set is about matching players and subjects. The target is the feature "like", which happens when a player likes a certain subject.

I used random forest and xgboost to deal with the imbalanced dataset. The models in the notebook are not yet fully optimized. Regardless, they should suffice for this exercise.

Next steps would be to try stacking, to see if there is a significant boost in AUC and accuracy.

About

My quick take on matching algorithms using RF, XGBoost on an imbalanced dataset.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published