ts-aggregator/project_template/DATrendNoSeasonality.cpp
2022-12-13 12:36:06 +04:00

126 lines
3.5 KiB
C++

//
// Ìåòîä ýêñïîíåíöèàëüíîãî ñãëàæèâàíèÿ è ïðîãíîçèðîâàíèÿ:
// äåìïèíã àääèòèâíûé òðåíä, áåç ñåçîííîñòè
//
#include "StdAfx.h"
#include <iostream>
#include "DATrendNoSeasonality.h"
#include "Param.h"
// êîíñòðóêòîð ñ çàäàííûìè íåïîñðåäñòâåííî ïàðàìåòðàìè
DATrendNoSeasonality::DATrendNoSeasonality(vector<double> timeSeries, int countPointForecast) {
this->x = timeSeries;
this->countPointForecast = countPointForecast;
this->partition();
}
DATrendNoSeasonality::~DATrendNoSeasonality() {
// îñâîáîæäàåòñÿ ïàìÿòü
std::vector<double> ().swap(S);
std::vector<double> ().swap(x);
std::vector<double> ().swap(T);
std::vector<double> ().swap(forecast);
}
// èíèöèàëèçàöèÿ ìîäåëè, çàäàíèå ïåðâîíà÷àëüíûõ çíà÷åíèé
void DATrendNoSeasonality::init() {
S.clear();
T.clear();
forecast.clear();
T.push_back(x[1] - x[0]);
S.push_back(x[0]);
forecast.push_back(S[0] + T[0] * phi);
}
// Çàäàòü ïàðàìåòð
void DATrendNoSeasonality::setParam(string paramName, double value) {
if (paramName.compare("alpha") == 0) {
this->alpha = value;
} else if (paramName.compare("gamma") == 0) {
this->gamma = value;
} else if (paramName.compare("phi") == 0) {
this->phi = value;
}
}
// ñôîðìèðîâàòü ìîäåëü
void DATrendNoSeasonality::createModel() {
this->init(); // èíèöèàëèçèðîâàòü ìîäåëü
double e = 0;
//âûïîëíÿåòñÿ ïðîõîä ìîäåëè ïî ñãëàæèâàíèþ è ïðîãíîçèðîâàíèþ countPointForecast òî÷åê
for (unsigned int t = 0; t < x.size()-1 + this->countPointForecast; t++) {
// ïîêà íå äîøëè äî êîíöà ðÿäà - ñãëàæèâàåì, èíà÷å ñòðîèì ïðîãíîç
if (t < x.size()) {
e = x[t]-forecast[t];
} else {
e = 0;
}
S.push_back(S[t] + phi * T[t] + alpha * e); // óðîâåíü
T.push_back(phi * T[t] + alpha * gamma * e); // çàòóõàþùèé òðåíä
forecast.push_back(S[t+1] + phi * T[t+1]); // ïðîãíîç
}
}
// ñôîðìèðîâàòü ìîäåëü
void DATrendNoSeasonality::createModelForEstimation() {
this->init(); // èíèöèàëèçèðîâàòü ìîäåëü
double e = 0;
//âûïîëíÿåòñÿ ïðîõîä ìîäåëè ïî ñãëàæèâàíèþ è ïðîãíîçèðîâàíèþ countPointForecast òî÷åê
for (unsigned int t = 0; t < xLearning.size()-1 + this->countPointForecast; t++) {
// ïîêà íå äîøëè äî êîíöà ðÿäà - ñãëàæèâàåì, èíà÷å ñòðîèì ïðîãíîç
if (t < xLearning.size()) {
e = xLearning[t]-forecast[t];
} else {
e = 0;
}
S.push_back(S[t] + phi * T[t] + alpha * e); // óðîâåíü
T.push_back(phi * T[t] + alpha * gamma * e); // çàòóõàþùèé òðåíä
forecast.push_back(S[t+1] + phi * T[t+1]); // ïðîãíîç
}
}
// ìåòîä ïîëó÷åíèÿ ïðîãíîçà
vector<double> DATrendNoSeasonality::getForecast() {
vector<double> result;
for (unsigned int i = forecast.size() - countPointForecast; i < forecast.size(); i++) {
result.push_back(forecast[i]);
}
return result;
}
// ìåòîä ïîëó÷åíèÿ îöåíêè ìîäåëè
double DATrendNoSeasonality::calcEstimation(Aic *aic) {
return aic->getValue(2, this->xEstimation, this->forecast);
}
// ìåòîä ïîëó÷åíèÿ îïòèìèçèðîâàííîãî çíà÷åíèÿ îäíîãî ïàðàìåòðà
// TODO: ðåàëèçîâàòü
Param* DATrendNoSeasonality::optimize(Estimation *est) {
Param *optimal = new Param();
double minSmape = 99999;
for (double al = 0.1; al < 1; al+= 0.05) {
for (double gam = 0.1; gam < 1; gam+= 0.05) {
for (double ph = 0.1; ph < 1; ph+= 0.05) {
this->setParam("alpha", al);
this->setParam("gamma", gam);
this->setParam("phi", ph);
this->createModelForEstimation();
double smapeValue = est->getValue(getXEstimation(), getForecast());
if (minSmape > smapeValue) {
minSmape = smapeValue;
optimal->alpha = al;
optimal->gamma = gam;
optimal->phi = phi;
}
}
}
}
return optimal;
}