matlab基于CNN-transformer的单特征时间序列预测

故障诊断与轴承RUL

于 2024-07-30 14:55:12 发布

阅读量24

点赞数

文章标签： cnn transformer 人工智能

本文链接：https://blog.csdn.net/2401_86544683/article/details/140796957

版权

框架：MATLAB R2023B以及以上

1.时间序列

2.网络模型

3.模型训练及预测情况

4.代码获取

感兴趣的朋友可以关注最后一行

% 参数设置
maxEpochs = 100;
miniBatchSize = 64;
learningRate = 0.001;
solver = 'adam';
shuffle = 'every-epoch';
gradientThreshold = 10;
executionEnvironment = "auto"; % chooses local GPU if available, otherwise CPU

options = trainingOptions(solver, ...
    'Plots','training-progress', ...
    'MaxEpochs', maxEpochs, ...
    'MiniBatchSize', miniBatchSize, ...
    'Shuffle', shuffle, ...
    'InitialLearnRate', learningRate, ...
    'GradientThreshold', gradientThreshold, ...
    'ExecutionEnvironment', executionEnvironment);

% 开始训练
net = trainNetwork(p_train, t_train, lgraph, options);

%% https://mbd.pub/o/bread/mbd-ZpmTlppt