1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
|
/*********************************************************************
MLDemos: A User-Friendly visualization toolkit for machine learning
Copyright (C) 2010 Basilio Noris
Contact: mldemos@b4silio.com
Evolution-Strategy Mixture of Logisitics Regression
Copyright (C) 2011 Stephane Magnenat
Contact: stephane at magnenat dot net
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public License,
version 3 as published by the Free Software Foundation.
This library is distributed in the hope that it will be useful, but
WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
Lesser General Public License for more details.
You should have received a copy of the GNU Lesser General Public
License along with this library; if not, write to the Free
Software Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*********************************************************************/
#include "interfaceESMLRClassifier.h"
#include "classifierESMLR.h"
#include "ui_paramsESMLR.h"
#include <cassert>
using namespace std;
ClassESMLR::ClassESMLR()
{
params = new Ui::ParametersESMLR();
params->setupUi(widget = new QWidget());
}
void ClassESMLR::SetParams(Classifier *classifier)
{
if(!classifier) return;
const u32 cutCount = params->esmlrCutSpin->value();
const float alpha = params->esmlrAlphaSpin->value();
const u32 genCount = params->esmlrGenSpin->value();
const u32 indPerDim = params->esmlrIndPerDimSpin->value();
ClassifierESMLR* c(dynamic_cast<ClassifierESMLR*>(classifier));
assert(c);
c->SetParams(cutCount, alpha, genCount, indPerDim);
}
QString ClassESMLR::GetAlgoString()
{
const u32 cutCount = params->esmlrCutSpin->value();
const float alpha = params->esmlrAlphaSpin->value();
const u32 genCount = params->esmlrGenSpin->value();
const u32 indPerDim = params->esmlrIndPerDimSpin->value();
return QString("ESMLR %1 %2 %3 %4").arg(cutCount).arg(alpha).arg(genCount).arg(indPerDim);
}
Classifier *ClassESMLR::GetClassifier()
{
ClassifierESMLR *classifier = new ClassifierESMLR();
SetParams(classifier);
return classifier;
}
void ClassESMLR::DrawModel(Canvas *canvas, QPainter &painter, Classifier *classifier)
{
if(!classifier || !canvas)
return;
painter.setRenderHint(QPainter::Antialiasing, true);
int posClass = 1;
FOR(i, canvas->data->GetCount())
{
const fvec sample = canvas->data->GetSample(i);
const int label = canvas->data->GetLabel(i);
const QPointF point = canvas->toCanvasCoords(canvas->data->GetSample(i));
const float response = classifier->Test(sample);
if(response > 0)
{
if(label == posClass)
Canvas::drawSample(painter, point, 9, 1);
else
Canvas::drawCross(painter, point, 6, 2);
}
else
{
if(label != posClass)
Canvas::drawSample(painter, point, 9, 0);
else
Canvas::drawCross(painter, point, 6, 0);
}
}
}
void ClassESMLR::SaveOptions(QSettings &settings)
{
settings.setValue("esmlrCut", params->esmlrCutSpin->value());
settings.setValue("esmlrAlpha", params->esmlrAlphaSpin->value());
settings.setValue("esmlrGen", params->esmlrGenSpin->value());
settings.setValue("esmlrIndPerDim", params->esmlrIndPerDimSpin->value());
}
bool ClassESMLR::LoadOptions(QSettings &settings)
{
if(settings.contains("esmlrCut")) params->esmlrCutSpin->setValue(settings.value("esmlrCut").toUInt());
if(settings.contains("esmlrAlpha")) params->esmlrAlphaSpin->setValue(settings.value("esmlrAlpha").toFloat());
if(settings.contains("esmlrGen")) params->esmlrGenSpin->setValue(settings.value("esmlrGen").toUInt());
if(settings.contains("esmlrIndPerDim")) params->esmlrIndPerDimSpin->setValue(settings.value("esmlrIndPerDim").toUInt());
return true;
}
void ClassESMLR::SaveParams(QTextStream &file)
{
file << "classificationOptions" << ":" << "esmlrCut" << " " << params->esmlrCutSpin->value() << "\n";
file << "classificationOptions" << ":" << "esmlrAlpha" << " " << params->esmlrAlphaSpin->value() << "\n";
file << "classificationOptions" << ":" << "esmlrGen" << " " << params->esmlrGenSpin->value() << "\n";
file << "classificationOptions" << ":" << "esmlrIndPerDim" << " " << params->esmlrIndPerDimSpin->value() << "\n";
}
bool ClassESMLR::LoadParams(QString name, float value)
{
if(name.endsWith("esmlrCut")) params->esmlrCutSpin->setValue((int)value);
if(name.endsWith("esmlrAlpha")) params->esmlrAlphaSpin->setValue(value);
if(name.endsWith("esmlrGen")) params->esmlrGenSpin->setValue((int)value);
if(name.endsWith("esmlrIndPerDim")) params->esmlrIndPerDimSpin->setValue((int)value);
return true;
}
|
