%% 从主函数中获取训练数据

trainD = evalin('base', 'trainD');

targetD = evalin('base', 'targetD');

ps_output = evalin('base', 'ps_output');

T_train = evalin('base', 'T_train');

or_dim= evalin('base', 'or_dim');

n_out = evalin('base', 'n_out');

%% 优化算法优化前,构建优化前的TCN模型

numChannels = or_dim;

maxPosition = 256*2;

numHeads = 4;

numKeyChannels = numHeads*32;

layers = [

sequenceInputLayer(numChannels,Name="input")

positionEmbeddingLayer(numChannels,maxPosition,Name="pos-emb");

additionLayer(2, Name="add")

智能算法及其模型预测

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐