sklearn使用小记GridSearchCV

最新推荐文章于 2022-03-11 09:30:07 发布

weixin_30736301

最新推荐文章于 2022-03-11 09:30:07 发布

阅读量174

点赞数

文章标签：人工智能数据结构与算法

原文链接：http://www.cnblogs.com/erbaodabao0611/p/9843196.html

版权

def test_grid_search():
    from sklearn import datasets,svm
    iris = datasets.load_iris()
    parameters = {'kernel': ('linear', 'rbf'), 'C': [1, 10,50,100]}  #注意score='roc_auc'是二分类的,多分类会报错
    svc = svm.SVC()
    clf = GridSearchCV(svc, parameters,cv=7)
    clf.fit(iris.data, iris.target)
    print(type(clf.best_params_))
    print(clf.best_params_)

转载于:https://www.cnblogs.com/erbaodabao0611/p/9843196.html