以下是使用 Python 构建支持向量机 (SVM) 模型的代码示例,该示例包括对用户输入值的合法性判断、模型训练和评估。

def build_clf3():
    global data, clf,X_train,X_test,Y_train,Y_test
    # 对用户输入值的合法性判断
    if data is None:
        messagebox.showerror('错误', '请先读取数据!')
        return
    
    # 创建支持向量机模型
    clf = svm.SVC(kernel='linear', C=1)

    # 训练模型
    clf.fit(X_train, Y_train)
    Y_pred = clf.predict(X_test)
    accuracy = accuracy_score(Y_test, Y_pred)
    precision = precision_score(Y_test, Y_pred, average='macro')
    recall = recall_score(Y_test, Y_pred, average='macro')
    f1 = f1_score(Y_test, Y_pred, average='macro')
    messagebox.showinfo('提示', f'模型构建成功!\n模型预测准确率:{accuracy:.2f}\n模型精确度:{precision:.2f}\n模型召回率:{recall:.2f}\n模型F1值:{f1:.2f}')

# 需要导入以下库:

```python
from sklearn import svm
from sklearn.metrics import accuracy_score, precision_score, recall_score, f1_score
from tkinter import messagebox

该代码示例使用 svm.SVC() 创建了一个线性 SVM 模型,并使用 fit() 方法进行训练。然后,使用测试集评估模型性能,并使用 accuracy_score(), precision_score(), recall_score()f1_score() 函数计算准确率、精确度、召回率和 F1 值。最后,使用 messagebox.showinfo() 将模型信息显示给用户。

Python 支持向量机 (SVM) 模型构建与评估

原文地址: http://www.cveoy.top/t/topic/oCK8 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录