GRU分类模型：基于PyTorch的数据集分析

以下是使用PyTorch编写GRU分类模型的详细代码：

import torch
import torch.nn as nn
import torch.optim as optim
from torch.utils.data import Dataset, DataLoader

# 定义自定义数据集类
class MyDataset(Dataset):
    def __init__(self, file_path):
        self.data = []
        self.labels = []
        with open(file_path, 'r') as f:
            for line in f:
                line = line.strip().split(',')
                self.data.append([float(x) for x in line[:-1]])
                self.labels.append(int(line[-1]))
    
    def __len__(self):
        return len(self.data)
    
    def __getitem__(self, index):
        return torch.FloatTensor(self.data[index]), self.labels[index]

# 定义GRU分类模型
class GRUClassifier(nn.Module):
    def __init__(self, input_size, hidden_size, num_layers, num_classes):
        super(GRUClassifier, self).__init__()
        self.hidden_size = hidden_size
        self.num_layers = num_layers
        self.gru = nn.GRU(input_size, hidden_size, num_layers, batch_first=True)
        self.fc = nn.Linear(hidden_size, num_classes)
    
    def forward(self, x):
        h0 = torch.zeros(self.num_layers, x.size(0), self.hidden_size).to(x.device)
        out, _ = self.gru(x, h0)
        out = self.fc(out[:, -1, :])
        return out

# 定义超参数
input_size = 22 # 输入特征维度
hidden_size = 128 # GRU隐藏层维度
num_layers = 2 # GRU层数
num_classes = 8 # 分类类别数
batch_size = 32 # 批量大小
num_epochs = 10 # 训练轮数
learning_rate = 0.001 # 学习率

# 创建数据集和数据加载器
dataset = MyDataset('data.txt')
dataloader = DataLoader(dataset, batch_size=batch_size, shuffle=True)

# 创建模型和优化器
model = GRUClassifier(input_size, hidden_size, num_layers, num_classes)
optimizer = optim.Adam(model.parameters(), lr=learning_rate)
criterion = nn.CrossEntropyLoss()

# 训练模型
for epoch in range(num_epochs):
    for data, labels in dataloader:
        data = data.unsqueeze(1) # 增加维度以适应GRU的输入
        outputs = model(data)
        loss = criterion(outputs, labels)
        
        optimizer.zero_grad()
        loss.backward()
        optimizer.step()
        
    print(f'Epoch [{epoch+1}/{num_epochs}], Loss: {loss.item()}')

# 保存模型
torch.save(model.state_dict(), 'model.pth')

请根据您的实际需求修改超参数和数据集路径，并确保数据集文件data.txt与代码文件在同一目录下。此代码将训练GRU分类模型，并将模型保存到model.pth文件中。