% 加载数据
data = load(‘classification_data.mat’);
X = data.X; % 特征数据
Y = data.Y; % 类别标签
% 划分训练集和测试集
train_ratio = 0.8; % 训练集比例
train_size = floor(train_ratio * size(X, 1));
train_X = X(1:train_size, 😃;
train_Y = categorical(Y(1:train_size));
test_X = X(train_size+1:end, 😃;
test_Y = categorical(Y(train_size+1:end));
% 构建Transformer-GRU模型
numFeatures = size(X, 2);
numClasses = numel(categories(Y));
layers = [
sequenceInputLayer(numFeatures)
transformerEncoderLayer(‘NumHeads’, 8, ‘ModelSize’, 64, ‘FeedForwardSize’, 128)
gruLayer(64, ‘OutputMode’, ‘last’)
fullyConnectedLayer(numClasses)
softmaxLayer
classificationLayer];
% 设置训练选项
options = trainingOptions(‘adam’, …
‘MaxEpochs’, 50, …
‘MiniBatchSize’, 32, …
‘Shuffle’, ‘every-epoch’, …
‘Plots’, ‘training-progress’);
% 训练模型
model = trainNetwork(train_X, train_Y, layers, options);
% 预测测试集数据
predictions = classify(model, test_X);
% 评估模型性能
accuracy = sum(predictions == test_Y) / numel(test_Y);
fprintf(‘准确率: %.2f%%\n’, accuracy * 100);
% 保存模型
save(‘classification_model.mat’, ‘model’);