使用data.csv文件中的第二列和第三列进行svm算法的python代码是什么
时间: 2024-05-13 20:19:05 浏览: 88
以下是使用Python中的Scikit-learn库来实现SVM算法的示例代码,假设data.csv文件中第二列为特征X,第三列为标签Y:
```python
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.svm import SVC
from sklearn.metrics import accuracy_score
# 读入CSV文件
data = pd.read_csv('data.csv')
# 将数据集划分为训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(data.iloc[:, 1:3], data.iloc[:, 3], test_size=0.3, random_state=42)
# 定义SVM模型
svm_model = SVC(kernel='linear', C=1)
# 训练SVM模型
svm_model.fit(X_train, y_train)
# 在测试集上评估模型性能
y_pred = svm_model.predict(X_test)
accuracy = accuracy_score(y_test, y_pred)
print('Accuracy:', accuracy)
```
此代码将数据集划分为训练集和测试集,使用线性核函数和正则化参数C=1定义SVM模型,并在训练集上进行训练。然后,使用训练好的模型预测测试集的标签,并计算分类准确度作为模型性能的评估指标。
阅读全文