python - AUC 的网格搜索查找参数

Question

我正在尝试为我的 SVM 找到参数，这些参数会给我最好的 AUC。但是我在 sklearn 中找不到 AUC 的任何评分函数。有人有想法吗？这是我的代码:

    parameters = {"C":[0.1, 1, 10, 100, 1000], "gamma":[0.1, 0.01, 0.001, 0.0001, 0.00001]}
    clf = SVC(kernel = "rbf")
    clf = GridSearchCV(clf, parameters, scoring = ???)
    svr.fit(features_train , labels_train)
    print svr.best_params_

那我可以用来做什么？？？获得高 AUC 分数的最佳参数？

Answer 1

您可以简单地使用:

clf = GridSearchCV(clf, parameters, scoring='roc_auc')