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