sklearn GridSearchCV中的refit是什么意思

  统计/机器学习 模型验证 Python    浏览次数:8665        分享
1

GridSearchCV中有个参数refit,默认值是True

class sklearn.model_selection.GridSearchCV(estimator, param_grid, scoring=None, fit_params=None, 
n_jobs=1, iid=True, refit=True, cv=None, verbose=0, pre_dispatch=‘2*n_jobs’, 
error_score=’raise’, return_train_score=’warn’)

这个refit什么意思,怎么理解?


 

TheTheThe   2017-10-22 23:59



   1个回答 
4

GridSearchCV是一个调参的过程,最后会告诉你最好的参数配置。

refit=True的意思就是GridSearchCV会按照最佳的参数配置重新训练模型。

clf = GridSearchCV(model, parameters, scoring='roc_auc', refit=True)
clf.predict(x_test)

在refit=True的时候,得到的clf是一个真正的模型,具有predict或者predict_proba的方法。


如果refit=False

clf = GridSearchCV(model, parameters, scoring='roc_auc', refit=False)

那么得到的clf并不是个真正的模型,不能进行predict。

SofaSofa数据科学社区DS面试题库 DS面经

染盘   2017-10-24 09:50



  相关讨论

sklearn cross_val_score怎么同时对多个scoring进行验证

调用sklearn中的classification_report,ValueError: Mix type of y not allowed, got types set(['binary', 'continuous'])

sklearn计算MAPE

sklearn cross_val_score中的参数pre_dispatch

sklearn.model_selection.cross_val_predict怎么固定random_state?

sklearn无法加载kfold

sklearn classification_report里的support是什么意思

关于sklearn.model_selection.PredefinedSplit的用法

怎么自定义sklearn GridSearchCV中评估函数的阈值

sklearn r2_score返回负数

  随便看看

二元分类问题中经常提到的TP,TN,FN,FP都是什么意思?

怎么把dataframe的一列转成整数类型

Resnet-18, Resnet-50, Resnet-101这些模型里的数字是什么意思?

python(matplotlib)中如何把折线图和柱状图画在一起?

python去掉中文文本中所有的标点符号