帮我用pytorch写一个将NSL-KDD进行one-hot编码的代码

以下是一个简单的PyTorch代码，将NSL-KDD进行one-hot编码：

import torch
import numpy as np

# 加载NSL-KDD数据集
train_data = np.loadtxt('./path/to/train_data.csv', delimiter=',')
test_data = np.loadtxt('./path/to/test_data.csv', delimiter=',')

# 将数据集拆分为X和y
X_train = train_data[:, :-1]
y_train = train_data[:, -1]
X_test = test_data[:, :-1]
y_test = test_data[:, -1]

# 进行one-hot编码
X_train = torch.Tensor(X_train).long()
X_train = torch.zeros(X_train.size(0), 122).scatter_(1, X_train, 1)
y_train = torch.Tensor(y_train).long()
y_train = torch.zeros(y_train.size(0), 5).scatter_(1, y_train.view(-1, 1), 1)

X_test = torch.Tensor(X_test).long()
X_test = torch.zeros(X_test.size(0), 122).scatter_(1, X_test, 1)
y_test = torch.Tensor(y_test).long()
y_test = torch.zeros(y_test.size(0), 5).scatter_(1, y_test.view(-1, 1), 1)

# 将数据集转换为PyTorch Dataset类型
train_dataset = torch.utils.data.TensorDataset(X_train, y_train)
test_dataset = torch.utils.data.TensorDataset(X_test, y_test)

上述代码中的第一行加载了NSL-KDD数据集。接下来，将数据集拆分为`X`和`y`。我们将`X`转换为长整型张量，并通过`scatter_`函数使用`one-hot`编码进行编码。`y`也会使用`scatter_`函数进行编码。最后，将编码后的张量转换为`PyTorch Dataset`类型。

注意，上述代码中的`122`和`5`分别是`X`和`y`的编码维度，这取决于数据集和类别数量。因此，在不同的数据集上使用此代码时，可能需要根据特定数据集的要求更改这些值。