2016-05-02 5 views
1

J'essaie de trouver un ensemble de paramètres optimal pour un XGB_Classifier en utilisant GridSearchCV. Étant donné que mes données sont très déséquilibrées, l'ajustement et l'évaluation (dans cross_validation) doivent être effectuées en utilisant des poids, donc je dois utiliser un marqueur personnalisé, qui prend comme paramètre un vecteur 'poids'. Cependant, je n'arrive pas à trouver un moyen pour que GridSearchCV transmette le vecteur 'poids' à un marqueur.GridSearchCV: passer des poids à un marqueur

Il y a eu quelques tentatives d'ajouter cette fonctionnalité à gridsearch:

https://github.com/ndawe/scikit-learn/commit/3da7fb708e67dd27d7ef26b40d29447b7dc565d7

Mais ils ne sont pas fusionnés en maître et maintenant je crains que ce code n'est pas compatible avec les changements en amont.

Est-ce que quelqu'un a fait face à un problème similaire et y a-t-il un moyen facile de s'en sortir?

Répondre