Matlab 深度学习工具箱案例学习与测试————求二阶微分方程

micro_xx

已于 2024-11-21 19:31:40 修改

阅读量707

点赞数 4

CC 4.0 BY-SA版权

分类专栏： Matlab深度学习文章标签： matlab 深度学习学习

于 2024-11-21 19:28:36 首次发布

本文链接：https://blog.youkuaiyun.com/qq_33268208/article/details/143951380

clc
clear

% 定义输入变量
x = linspace(0,2,10000)';

% 定义网络的层参数
inputSize = 1;
layers = [
    featureInputLayer(inputSize,Normalization="none")
    fullyConnectedLayer(10)
    sigmoidLayer
    fullyConnectedLayer(1)
    sigmoidLayer];
% 创建网络
net = dlnetwork(layers);

% 训练轮数
numEpochs = 15;
% 每个Batch的数据个数
miniBatchSize = 100;

% SGDM优化方法设置的参数
initialLearnRate = 0.5;
learnRateDropFactor = 0.5;
learnRateDropPeriod = 5;
momentum = 0.9;
velocity = [];

% 损失函数里面考虑初始条件的系数
icCoeff = 7;

% ArrayDatastore
ads = arrayDatastore(x,IterationDimension=1);
% 创建一个用于处理管理深度学习数据的对象
mbq = minibatchqueue(ads, ...
    MiniBatchSize=miniB