【发布时间】:2018-10-10 14:01:45
【问题描述】:
我曾经创建循环来为我的模型寻找最佳参数,这增加了我的编码错误,所以我决定使用GridSearchCV。
我正在尝试为我的模型找出 PCA 的最佳参数(我想要进行网格搜索的唯一参数)。
在这个模型中,归一化后,我想将原始特征与 PCA 简化特征结合起来,然后应用线性 SVM。
然后我保存整个模型来预测我的输入。
我尝试拟合数据的行中有错误,因此我可以使用best_estimator_ 和best_params_ 函数。
错误说:TypeError: The score function should be a callable, all (<type 'str'>) was passed. 我没有使用任何可能需要在GridSearchCV 中提供字符串的参数,所以不知道为什么会出现此错误
我还想知道在保存我的模型之前print("shape after model",X.shape) 行是否应该根据所有可能的参数打印(150, 7) and (150, 5)?
from sklearn.pipeline import Pipeline, FeatureUnion
from sklearn.model_selection import GridSearchCV
from sklearn.svm import SVC
from sklearn.datasets import load_iris
from sklearn.decomposition import PCA
from sklearn.feature_selection import SelectKBest
from sklearn.preprocessing import StandardScaler
from sklearn.externals import joblib
from numpy import array
iris = load_iris()
X, y = iris.data, iris.target
print(X.shape) #prints (150, 4)
print (y)
#cretae models and piplline them
combined_features = FeatureUnion([("pca", PCA()), ("univ_select", SelectKBest(k='all'))])
svm = SVC(kernel="linear")
pipeline = Pipeline([("scale", StandardScaler()),("features", combined_features), ("svm", svm)])
# Do grid search over n_components:
param_grid = dict(features__pca__n_components=[1,3])
grid_search = GridSearchCV(pipeline, param_grid=param_grid, cv=5, verbose=10)
grid_search.fit(X, y)
print("best parameters", grid_search.best_params_)
print("shape after model",X.shape) #should this print (150, 7) or (150, 5) based on best parameter?
#save the model
joblib.dump(grid_search.best_estimator_, 'model.pkl', compress = 1)
#new data to predict
Input=[ 2.9 , 4. ,1.2 ,0.2]
Input= array(Input)
#use the saved model to predict the new data
modeltrain="model.pkl"
modeltrain_saved = joblib.load(modeltrain)
model_predictions = modeltrain_saved.predict(Input.reshape(1, -1))
print(model_predictions)
我根据答案更新了代码
【问题讨论】:
标签: python scikit-learn svm pca grid-search