支持向量机(SVM)是一种强大的监督学习模型,主要用于分类问题,但也可以用于回归问题,这时称为支持向量机回归(SVR)。SVR通过找到一个超平面来拟合数据点,使得所有数据点到该超平面的距离(即误差)不超过一个给定的阈值。
以下是一个简单的SVR模型训练和预测的示例代码:
from sklearn.svm import SVR
from sklearn.model_selection import train_test_split
from sklearn.metrics import mean_squared_error
import numpy as np
# 假设我们有一些数据
X = np.sort(5 * np.random.rand(80, 1), axis=0)
y = np.sin(X).ravel() + np.random.randn(80) * 0.1
# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
# 创建SVR模型,这里使用RBF核
svr_model = SVR(kernel='rbf', C=1e3, gamma=0.1)
# 训练模型
svr_model.fit(X_train, y_train)
# 预测测试集
y_pred = svr_model.predict(X_test)
# 计算均方误差
mse = mean_squared_error(y_test, y_pred)
print(f"Mean Squared Error: {mse}")
通过以上信息,你应该能够对支持向量机回归有一个全面的了解,并能够应用它来解决实际问题。
领取专属 10元无门槛券
手把手带您无忧上云