Warrior_EA/references/MQL5/Experts/CIC/Pretrain.mq5

<EFBFBD><EFBFBD>//+------------------------------------------------------------------+

//|                                                        Study.mq5 |

//|                                                   Copyright DNG<EFBFBD> |

//|                                https://www.mql5.com/en/users/dng |

//+------------------------------------------------------------------+

#property copyright "Copyright DNG<00>"

#property link      "https://www.mql5.com/en/users/dng"

#property version   "1.00"

//+------------------------------------------------------------------+

//|                                                                  |

//+------------------------------------------------------------------+

#define Study

#include "Trajectory.mqh"

//+------------------------------------------------------------------+

//| Input parameters                                                 |

//+------------------------------------------------------------------+

input int                  Iterations     = 100000;

input float                Tau            = 0.001f;

//+------------------------------------------------------------------+

//|                                                                  |

//+------------------------------------------------------------------+

STrajectory          Buffer[];

CNet                 Encoder;

CNet                 TargetEncoder;

CNet                 Actor;

CNet                 Critic1;

CNet                 Critic2;

CNet                 Convolution;

CNet                 Descriminator;

CNet                 SkillProject;

//---

float                dError;

datetime             dtStudied;

//---

CBufferFloat         State;

CBufferFloat         Account;

CBufferFloat         TargetState;

CBufferFloat         TargetAccount;

CBufferFloat         Actions;

CBufferFloat         Gradient;

CBufferFloat         Skills;

CBufferFloat         *Result;

vector<float>        check;

//---

COpenCLMy           *OpenCL;

//+------------------------------------------------------------------+

//| Expert initialization function                                   |

//+------------------------------------------------------------------+

int OnInit()

  {

//---

   ResetLastError();

   if(!LoadTotalBase())

     {

      PrintFormat("Error of load study data: %d", GetLastError());

      return INIT_FAILED;

     }

//--- load models

   float temp;

   if(!Encoder.Load(FileName + "Enc.nnw", temp, temp, temp, dtStudied, true) ||

      !Actor.Load(FileName + "Act.nnw", temp, temp, temp, dtStudied, true) ||

      !Critic1.Load(FileName + "Crt1.nnw", temp, temp, temp, dtStudied, true) ||

      !Critic2.Load(FileName + "Crt2.nnw", temp, temp, temp, dtStudied, true) ||

      !Descriminator.Load(FileName + "Des.nnw", temp, temp, temp, dtStudied, true) ||

      !SkillProject.Load(FileName + "Skp.nnw", temp, temp, temp, dtStudied, true) ||

      !Convolution.Load(FileName + "CNN.nnw", temp, temp, temp, dtStudied, true) ||

      !TargetEncoder.Load(FileName + "Enc.nnw", temp, temp, temp, dtStudied, true))

     {

      CArrayObj *encoder = new CArrayObj();

      CArrayObj *actor = new CArrayObj();

      CArrayObj *critic = new CArrayObj();

      CArrayObj *descrim = new CArrayObj();

      CArrayObj *convolution = new CArrayObj();

      CArrayObj *skill_poject = new CArrayObj();

      if(!CreateDescriptions(encoder,actor, critic, convolution,descrim,skill_poject))

        {

         delete encoder;

         delete actor;

         delete critic;

         delete descrim;

         delete convolution;

         delete skill_poject;

         return INIT_FAILED;

        }

      if(!Encoder.Create(encoder) || !Actor.Create(actor) ||

         !Critic1.Create(critic) || !Critic2.Create(critic) ||

         !Descriminator.Create(descrim) || !SkillProject.Create(skill_poject) ||

         !Convolution.Create(convolution))

        {

         delete encoder;

         delete actor;

         delete critic;

         delete descrim;

         delete convolution;

         delete skill_poject;

         return INIT_FAILED;

        }

      if(!TargetEncoder.Create(encoder))

        {

         delete encoder;

         delete actor;

         delete critic;

         delete descrim;

         delete convolution;

         delete skill_poject;

         return INIT_FAILED;

        }

      delete encoder;

      delete actor;

      delete critic;

      delete descrim;

      delete convolution;

      delete skill_poject;

      //---

      TargetEncoder.WeightsUpdate(GetPointer(Encoder), 1.0f);

     }

//---

   OpenCL = Actor.GetOpenCL();

   Encoder.SetOpenCL(OpenCL);

   Critic1.SetOpenCL(OpenCL);

   Critic2.SetOpenCL(OpenCL);

   TargetEncoder.SetOpenCL(OpenCL);

   Descriminator.SetOpenCL(OpenCL);

   SkillProject.SetOpenCL(OpenCL);

   Convolution.SetOpenCL(OpenCL);

//---

   vector<float> ActorResult;

   Actor.getResults(ActorResult);

   if(ActorResult.Size() != NActions)

     {

      PrintFormat("The scope of the actor does not match the actions count (%d <> %d)", NActions, Result.Total());

      return INIT_FAILED;

     }

//---

   Encoder.GetLayerOutput(0, Result);

   if(Result.Total() != (HistoryBars * BarDescr))

     {

      PrintFormat("Input size of State Encoder doesn't match state description (%d <> %d)", Result.Total(), (HistoryBars * BarDescr));

      return INIT_FAILED;

     }

//---

   vector<float> EncoderResults;

   Actor.GetLayerOutput(0,Result);

   Encoder.getResults(EncoderResults);

   if(Result.Total() != int(EncoderResults.Size()))

     {

      PrintFormat("Input size of Actor doesn't match Encoder outputs (%d <> %d)", Result.Total(), EncoderResults.Size());

      return INIT_FAILED;

     }

//---

   Actor.GetLayerOutput(LatentLayer, Result);

   int latent_state = Result.Total();

   Critic1.GetLayerOutput(0, Result);

   if(Result.Total() != latent_state)

     {

      PrintFormat("Input size of Critic doesn't match latent state Actor (%d <> %d)", Result.Total(), latent_state);

      return INIT_FAILED;

     }

//---

   Gradient.BufferInit(MathMax(AccountDescr,NSkills), 0);

//---

   if(!EventChartCustom(ChartID(), 1, 0, 0, "Init"))

     {

      PrintFormat("Error of create study event: %d", GetLastError());

      return INIT_FAILED;

     }

//---

   return(INIT_SUCCEEDED);

  }

//+------------------------------------------------------------------+

//| Expert deinitialization function                                 |

//+------------------------------------------------------------------+

void OnDeinit(const int reason)

  {

//---

   TargetEncoder.WeightsUpdate(GetPointer(Encoder), Tau);

   Actor.Save(FileName + "Act.nnw", 0, 0, 0, TimeCurrent(), true);

   TargetEncoder.Save(FileName + "Enc.nnw", Critic1.getRecentAverageError(), 0, 0, TimeCurrent(), true);

   Critic1.Save(FileName + "Crt1.nnw", Critic1.getRecentAverageError(), 0, 0, TimeCurrent(), true);

   Critic2.Save(FileName + "Crt2.nnw", Critic2.getRecentAverageError(), 0, 0, TimeCurrent(), true);

   Convolution.Save(FileName + "CNN.nnw", 0, 0, 0, TimeCurrent(), true);

   Descriminator.Save(FileName + "Des.nnw", 0, 0, 0, TimeCurrent(), true);

   SkillProject.Save(FileName + "Skp.nnw", 0, 0, 0, TimeCurrent(), true);

   delete Result;

  }

//+------------------------------------------------------------------+

//| ChartEvent function                                              |

//+------------------------------------------------------------------+

void OnChartEvent(const int id,

                  const long &lparam,

                  const double &dparam,

                  const string &sparam)

  {

//---

   if(id == 1001)

      Train();

  }

//+------------------------------------------------------------------+

//| Train function                                                   |

//+------------------------------------------------------------------+

void Train(void)

  {

   int total_tr = ArraySize(Buffer);

   uint ticks = GetTickCount();

//---

   int total_states = Buffer[0].Total - 1;

   for(int i = 1; i < total_tr; i++)

      total_states += Buffer[i].Total - 1;

   vector<float> temp;

   Convolution.getResults(temp);

   matrix<float> state_embedding = matrix<float>::Zeros(total_states,temp.Size());

   int state = 0;

   for(int tr = 0; tr < total_tr; tr++)

     {

      for(int st = 0; st < Buffer[tr].Total - 1; st++)

        {

         State.AssignArray(Buffer[tr].States[st].state);

         float PrevBalance = Buffer[tr].States[MathMax(st,0)].account[0];

         float PrevEquity = Buffer[tr].States[MathMax(st,0)].account[1];

         State.Add((Buffer[tr].States[st].account[0] - PrevBalance) / PrevBalance);

         State.Add(Buffer[tr].States[st].account[1] / PrevBalance);

         State.Add((Buffer[tr].States[st].account[1] - PrevEquity) / PrevEquity);

         State.Add(Buffer[tr].States[st].account[2]);

         State.Add(Buffer[tr].States[st].account[3]);

         State.Add(Buffer[tr].States[st].account[4] / PrevBalance);

         State.Add(Buffer[tr].States[st].account[5] / PrevBalance);

         State.Add(Buffer[tr].States[st].account[6] / PrevBalance);

         double x = (double)Buffer[tr].States[st].account[7] / (double)(D'2024.01.01' - D'2023.01.01');

         State.Add((float)MathSin(x != 0 ? 2.0 * M_PI * x : 0));

         x = (double)Buffer[tr].States[st].account[7] / (double)PeriodSeconds(PERIOD_MN1);

         State.Add((float)MathCos(x != 0 ? 2.0 * M_PI * x : 0));

         x = (double)Buffer[tr].States[st].account[7] / (double)PeriodSeconds(PERIOD_W1);

         State.Add((float)MathSin(x != 0 ? 2.0 * M_PI * x : 0));

         x = (double)Buffer[tr].States[st].account[7] / (double)PeriodSeconds(PERIOD_D1);

         State.Add((float)MathSin(x != 0 ? 2.0 * M_PI * x : 0));

         //---

         State.AddArray(Buffer[tr].States[st + 1].state);