svm pytorch

在PyTorch中，可以通过定义一个类来实现支持向量机（SVM）。下面是一个基于PyTorch的SVM的示例代码：

import torch
import torch.nn as nn
import torch.optim as optim

class SVM(nn.Module):
    def __init__(self, input_size, num_classes):
        super(SVM, self).__init__()
        self.linear = nn.Linear(input_size, num_classes)

    def forward(self, x):
        return self.linear(x)

# 定义训练数据
x_train = torch.tensor([[1., 1.], [-1., 1.], [-1., -1.], [1., -1.]])
y_train = torch.tensor([1., -1., -1., -1.])

# 定义SVM模型
svm = SVM(input_size=2, num_classes=1)
criterion = nn.HingeEmbeddingLoss()
optimizer = optim.SGD(svm.parameters(), lr=0.01)

# 训练模型
num_epochs = 1000
for epoch in range(num_epochs):
    optimizer.zero_grad()
    outputs = svm(x_train)
    loss = criterion(outputs.squeeze(), y_train)
    loss.backward()
    optimizer.step()

# 测试模型
x_test = torch.tensor([[2., 2.], [-2., 2.], [-2., -2.], [2., -2.]])
outputs = svm(x_test)
predicted = torch.sign(outputs).squeeze()
print(predicted)

在上述代码中，我们首先定义了一个SVM类，它包括一个线性层和一个前向传递方法。然后我们定义了训练数据x_train和y_train。接着，我们实例化了SVM类，并定义了损失函数和优化器。在训练过程中，我们使用`optimizer.zero_grad()`方法清除梯度，计算输出和损失，然后通过调用`optimizer.step()`方法更新参数。最后，我们使用x_test测试数据进行预测，并将输出通过`torch.sign()`方法转换为-1或1的值。