2016-05-02 11 views
1

Ich versuche mit GridSearchCV einen optimalen Parametersatz für einen XGB_Classifier zu finden. Da meine Daten sehr unausgewogen sind, müssen sowohl das Fitting als auch das Scoring (in cross_validation) mit Hilfe von Gewichtungen durchgeführt werden. Daher muss ich einen benutzerdefinierten Scorer verwenden, der einen "Gewichtungsvektor" als Parameter verwendet. Allerdings kann ich keinen Weg finden, wie GridSearchCV den Vektor "Gewichtungen" an einen Scorer übergibt.GridSearchCV: Gewichte an einen Scorer übergeben

Es gab einige Versuche, diese Funktionalität hinzufügen, um Gridsearch: in Master

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

Aber sie waren nicht verschmolzen, und ich bin jetzt Angst, dass dieser Code mit Upstream-Änderungen nicht kompatibel ist.

Hat jemand ein ähnliches Problem gehabt und gibt es einen "einfachen" Weg damit umzugehen?

Antwort