sklearn中gridsearchcv 与pipeline结合使用

最新推荐文章于 2022-07-10 11:43:52 发布

小小哆啦的口袋

最新推荐文章于 2022-07-10 11:43:52 发布

阅读量1.6k

点赞数 2

分类专栏：机器学习文章标签： MLearning

本文链接：https://blog.csdn.net/nowfuture/article/details/81915341

版权

机器学习专栏收录该内容

5 篇文章 0 订阅

订阅专栏

X = train[column]
y=(train["class"]-1).astype(int)
tfid = TfidfVectorizer(use_idf = 1,
                       smooth_idf = 1,
                       sublinear_tf = 1,
                       max_df = 0.9,
                       min_df = 3,
                       ngram_range = (1,3))
X_feature = tfid.fit_transform(X)

esta = svm.LinearSVC()
pip = Pipeline([('tfid',tfid),('esta',esta)])
param = dict(tfid__max_features=[100000,200000,300000])
grid = GridSearchCV(pip,param_grid = param,scoring='f1_weighted')
grid.fit(X,y)
print('best params:'% grid.best_params_)

print("Best score: %0.3f" % grid.best_score_)