我试图使用 GridSearchCV 来进行 SVM 线性分类,但得到了以下错误:
AttributeError: 'SVC' object has no attribute 'best_estimator_'
SVM 线性分类的代码如下:
classifier = SVC()classifier = GridSearchCV(classifier, {'C':[0.001, 0.01, 0.1, 1, 10,0.1, 100, 1000]}, cv=3, n_jobs=4)classifier = SVC(kernel='linear')classifier.fit(train_vectors, train_labels)classifier = classifier.best_estimator_
谁能帮帮我?
回答:
这样做:
classifier = SVC(kernel='linear')gridsearch = GridSearchCV(classifier, {'C':[0.001, 0.01, 0.1, 1, 10,0.1, 100, 1000]}, cv=3, n_jobs=4)gridsearch.fit(train_vectors, train_labels)best_params = gridsearch.best_params_classifier = gridsearch.best_estimator_