Python超参数自动搜索模块GridSearchCV

最新推荐文章于 2023-04-28 09:41:57 发布

淮南草

最新推荐文章于 2023-04-28 09:41:57 发布

阅读量781

点赞数

分类专栏：机器学习预处理

本文链接：https://blog.csdn.net/zhuisaozhang1292/article/details/81456335

版权

本文详细介绍了Python中用于超参数优化的GridSearchCV模块，包括其工作原理、使用方法及如何结合交叉验证提升模型性能。通过实例解析，帮助读者掌握如何在机器学习项目中有效地进行超参数调优。

摘要由CSDN通过智能技术生成

from __future__ import print_function

from pprint import pprint
from time import time
import logging

from sklearn.datasets import fetch_20newsgroups
from sklearn.feature_extraction.text import CountVectorizer
from sklearn.feature_extraction.text import TfidfTransformer
from sklearn.linear_model import SGDClassifier
from sklearn.model_selection import GridSearchCV
from sklearn.pipeline import Pipeline

print(__doc__)
 
# Display progress logs on stdout
logging.basicConfig(level=logging.INFO,
                    format='%(asctime)s %(levelname)s %(message)s')
 
 
# #############################################################################
# Load some categories from the training set
categories = [
    'alt.atheism',
    'talk.religion.misc',
]
# Uncomment the following to do the analysis on all the categories
#categories = None
 
print("Loading 20 newsgroups dataset for categories:")
print