保存并加载SVM训练的moedel

2024-09-28 01:33:27 发布

您现在位置:Python中文网/ 问答频道 /正文

我已经训练了一个SVM分类器,并且得到了很好的结果,现在我需要保存该模型,在需要的时候加载它,并预测新的看不见的数据。 这是我的代码我应该添加什么

from sklearn.model_selection import train_test_split
from sklearn.feature_selection import f_classif
from sklearn.model_selection import GridSearchCV
from sklearn.feature_selection import SelectKBest
import pandas as pd
from sklearn.svm import SVC
from sklearn.metrics import accuracy_score
from sklearn.metrics import classification_report, confusion_matrix


param_grid = {'C': [0.1,1, 10, 100], 'gamma': [1,0.1,0.01,0.001],'kernel': ['linear','rbf', 'poly', 'sigmoid']}

def select_features(X_train,y_train,X_test):
     fs = SelectKBest(score_func=f_classif, k='all')
     fs.fit(X_train, y_train)
     X_train_fs = fs.transform(X_train)
     X_test_fs  = fs.transform(X_test)
     return X_train_fs,X_test_fs,fs
 
data = pd.read_csv('d:/f.csv') 
d=data.values
X = d[:,0:207]
y = d[:,208]
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.33, random_state=1)
X_train_fs, X_test_fs, fs =select_features(X_train,y_train,X_test)
model = GridSearchCV(SVC(),param_grid,refit=True,verbose=2,n_jobs=-1)
model.fit(X_train_fs, y_train)
y_pred = model.predict(X_test_fs)
print('Best GridSearchCV parameters: ',model.best_params_)
print("Accuracy: ",accuracy_score(y_test, y_pred))
print(confusion_matrix(y_test,y_pred))
print(classification_report(y_test,y_pred))

Tags: fromtestimportmodeltrainsklearnfsfeature
1条回答
网友
1楼 · 发布于 2024-09-28 01:33:27

简短答复: SVChas没有loadsave函数。但是,参数存储在dict中,这意味着您可以将它们另存为json。之后,您可以加载dict并使用set_params方法重新实例化SVM

from sklearn.svm import SVC
svc = SVC()
svc.set_params(loaded_params)

这是你的问题吗

相关问题 更多 >

    热门问题