|
|
@@ -0,0 +1,437 @@ |
|
|
|
#include <QtGui> |
|
|
|
#include <stdio.h> |
|
|
|
#include <stdlib.h> |
|
|
|
#include <string.h> |
|
|
|
#include <ctype.h> |
|
|
|
#include <list> |
|
|
|
#include "../../svm.h" |
|
|
|
using namespace std; |
|
|
|
|
|
|
|
#define DEFAULT_PARAM "-t 2 -c 100" |
|
|
|
#define XLEN 500 |
|
|
|
#define YLEN 500 |
|
|
|
|
|
|
|
QRgb colors[] = |
|
|
|
{ |
|
|
|
qRgb(0,0,0), |
|
|
|
qRgb(0,120,120), |
|
|
|
qRgb(120,120,0), |
|
|
|
qRgb(120,0,120), |
|
|
|
qRgb(0,200,200), |
|
|
|
qRgb(200,200,0), |
|
|
|
qRgb(200,0,200) |
|
|
|
}; |
|
|
|
|
|
|
|
class SvmToyWindow : public QWidget |
|
|
|
{ |
|
|
|
|
|
|
|
Q_OBJECT |
|
|
|
|
|
|
|
public: |
|
|
|
SvmToyWindow(); |
|
|
|
~SvmToyWindow(); |
|
|
|
protected: |
|
|
|
virtual void mousePressEvent( QMouseEvent* ); |
|
|
|
virtual void paintEvent( QPaintEvent* ); |
|
|
|
|
|
|
|
private: |
|
|
|
QPixmap buffer; |
|
|
|
QPixmap icon1; |
|
|
|
QPixmap icon2; |
|
|
|
QPixmap icon3; |
|
|
|
QPushButton button_change_icon; |
|
|
|
QPushButton button_run; |
|
|
|
QPushButton button_clear; |
|
|
|
QPushButton button_save; |
|
|
|
QPushButton button_load; |
|
|
|
QLineEdit input_line; |
|
|
|
QPainter buffer_painter; |
|
|
|
struct point { |
|
|
|
double x, y; |
|
|
|
signed char value; |
|
|
|
}; |
|
|
|
list<point> point_list; |
|
|
|
int current_value; |
|
|
|
const QPixmap& choose_icon(int v) |
|
|
|
{ |
|
|
|
if(v==1) return icon1; |
|
|
|
else if(v==2) return icon2; |
|
|
|
else return icon3; |
|
|
|
} |
|
|
|
void clear_all() |
|
|
|
{ |
|
|
|
point_list.clear(); |
|
|
|
buffer.fill(Qt::black); |
|
|
|
repaint(); |
|
|
|
} |
|
|
|
void draw_point(const point& p) |
|
|
|
{ |
|
|
|
const QPixmap& icon = choose_icon(p.value); |
|
|
|
buffer_painter.drawPixmap((int)(p.x*XLEN),(int)(p.y*YLEN),icon); |
|
|
|
repaint(); |
|
|
|
} |
|
|
|
void draw_all_points() |
|
|
|
{ |
|
|
|
for(list<point>::iterator p = point_list.begin(); p != point_list.end();p++) |
|
|
|
draw_point(*p); |
|
|
|
} |
|
|
|
private slots: |
|
|
|
void button_change_icon_clicked() |
|
|
|
{ |
|
|
|
++current_value; |
|
|
|
if(current_value > 3) current_value = 1; |
|
|
|
button_change_icon.setIcon(choose_icon(current_value)); |
|
|
|
} |
|
|
|
void button_run_clicked() |
|
|
|
{ |
|
|
|
// guard |
|
|
|
if(point_list.empty()) return; |
|
|
|
|
|
|
|
svm_parameter param; |
|
|
|
int i,j; |
|
|
|
|
|
|
|
// default values |
|
|
|
param.svm_type = C_SVC; |
|
|
|
param.kernel_type = RBF; |
|
|
|
param.degree = 3; |
|
|
|
param.gamma = 0; |
|
|
|
param.coef0 = 0; |
|
|
|
param.nu = 0.5; |
|
|
|
param.cache_size = 100; |
|
|
|
param.C = 1; |
|
|
|
param.eps = 1e-3; |
|
|
|
param.p = 0.1; |
|
|
|
param.shrinking = 1; |
|
|
|
param.probability = 0; |
|
|
|
param.nr_weight = 0; |
|
|
|
param.weight_label = NULL; |
|
|
|
param.weight = NULL; |
|
|
|
|
|
|
|
// parse options |
|
|
|
const char *p = input_line.text().toAscii().constData(); |
|
|
|
|
|
|
|
while (1) { |
|
|
|
while (*p && *p != '-') |
|
|
|
p++; |
|
|
|
|
|
|
|
if (*p == '\0') |
|
|
|
break; |
|
|
|
|
|
|
|
p++; |
|
|
|
switch (*p++) { |
|
|
|
case 's': |
|
|
|
param.svm_type = atoi(p); |
|
|
|
break; |
|
|
|
case 't': |
|
|
|
param.kernel_type = atoi(p); |
|
|
|
break; |
|
|
|
case 'd': |
|
|
|
param.degree = atoi(p); |
|
|
|
break; |
|
|
|
case 'g': |
|
|
|
param.gamma = atof(p); |
|
|
|
break; |
|
|
|
case 'r': |
|
|
|
param.coef0 = atof(p); |
|
|
|
break; |
|
|
|
case 'n': |
|
|
|
param.nu = atof(p); |
|
|
|
break; |
|
|
|
case 'm': |
|
|
|
param.cache_size = atof(p); |
|
|
|
break; |
|
|
|
case 'c': |
|
|
|
param.C = atof(p); |
|
|
|
break; |
|
|
|
case 'e': |
|
|
|
param.eps = atof(p); |
|
|
|
break; |
|
|
|
case 'p': |
|
|
|
param.p = atof(p); |
|
|
|
break; |
|
|
|
case 'h': |
|
|
|
param.shrinking = atoi(p); |
|
|
|
break; |
|
|
|
case 'b': |
|
|
|
param.probability = atoi(p); |
|
|
|
break; |
|
|
|
case 'w': |
|
|
|
++param.nr_weight; |
|
|
|
param.weight_label = (int *)realloc(param.weight_label,sizeof(int)*param.nr_weight); |
|
|
|
param.weight = (double *)realloc(param.weight,sizeof(double)*param.nr_weight); |
|
|
|
param.weight_label[param.nr_weight-1] = atoi(p); |
|
|
|
while(*p && !isspace(*p)) ++p; |
|
|
|
param.weight[param.nr_weight-1] = atof(p); |
|
|
|
break; |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
// build problem |
|
|
|
svm_problem prob; |
|
|
|
|
|
|
|
prob.l = point_list.size(); |
|
|
|
prob.y = new double[prob.l]; |
|
|
|
|
|
|
|
if(param.kernel_type == PRECOMPUTED) |
|
|
|
{ |
|
|
|
} |
|
|
|
else if(param.svm_type == EPSILON_SVR || |
|
|
|
param.svm_type == NU_SVR) |
|
|
|
{ |
|
|
|
if(param.gamma == 0) param.gamma = 1; |
|
|
|
svm_node *x_space = new svm_node[2 * prob.l]; |
|
|
|
prob.x = new svm_node *[prob.l]; |
|
|
|
|
|
|
|
i = 0; |
|
|
|
for (list <point>::iterator q = point_list.begin(); q != point_list.end(); q++, i++) |
|
|
|
{ |
|
|
|
x_space[2 * i].index = 1; |
|
|
|
x_space[2 * i].value = q->x; |
|
|
|
x_space[2 * i + 1].index = -1; |
|
|
|
prob.x[i] = &x_space[2 * i]; |
|
|
|
prob.y[i] = q->y; |
|
|
|
} |
|
|
|
|
|
|
|
// build model & classify |
|
|
|
svm_model *model = svm_train(&prob, ¶m); |
|
|
|
svm_node x[2]; |
|
|
|
x[0].index = 1; |
|
|
|
x[1].index = -1; |
|
|
|
int *j = new int[XLEN]; |
|
|
|
|
|
|
|
for (i = 0; i < XLEN; i++) |
|
|
|
{ |
|
|
|
x[0].value = (double) i / XLEN; |
|
|
|
j[i] = (int)(YLEN*svm_predict(model, x)); |
|
|
|
} |
|
|
|
|
|
|
|
buffer_painter.setPen(colors[0]); |
|
|
|
buffer_painter.drawLine(0,0,0,YLEN-1); |
|
|
|
|
|
|
|
int p = (int)(param.p * YLEN); |
|
|
|
for(i = 1; i < XLEN; i++) |
|
|
|
{ |
|
|
|
buffer_painter.setPen(colors[0]); |
|
|
|
buffer_painter.drawLine(i,0,i,YLEN-1); |
|
|
|
|
|
|
|
buffer_painter.setPen(colors[5]); |
|
|
|
buffer_painter.drawLine(i-1,j[i-1],i,j[i]); |
|
|
|
|
|
|
|
if(param.svm_type == EPSILON_SVR) |
|
|
|
{ |
|
|
|
buffer_painter.setPen(colors[2]); |
|
|
|
buffer_painter.drawLine(i-1,j[i-1]+p,i,j[i]+p); |
|
|
|
|
|
|
|
buffer_painter.setPen(colors[2]); |
|
|
|
buffer_painter.drawLine(i-1,j[i-1]-p,i,j[i]-p); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
svm_free_and_destroy_model(&model); |
|
|
|
delete[] j; |
|
|
|
delete[] x_space; |
|
|
|
delete[] prob.x; |
|
|
|
delete[] prob.y; |
|
|
|
} |
|
|
|
else |
|
|
|
{ |
|
|
|
if(param.gamma == 0) param.gamma = 0.5; |
|
|
|
svm_node *x_space = new svm_node[3 * prob.l]; |
|
|
|
prob.x = new svm_node *[prob.l]; |
|
|
|
|
|
|
|
i = 0; |
|
|
|
for (list <point>::iterator q = point_list.begin(); q != point_list.end(); q++, i++) |
|
|
|
{ |
|
|
|
x_space[3 * i].index = 1; |
|
|
|
x_space[3 * i].value = q->x; |
|
|
|
x_space[3 * i + 1].index = 2; |
|
|
|
x_space[3 * i + 1].value = q->y; |
|
|
|
x_space[3 * i + 2].index = -1; |
|
|
|
prob.x[i] = &x_space[3 * i]; |
|
|
|
prob.y[i] = q->value; |
|
|
|
} |
|
|
|
|
|
|
|
// build model & classify |
|
|
|
svm_model *model = svm_train(&prob, ¶m); |
|
|
|
svm_node x[3]; |
|
|
|
x[0].index = 1; |
|
|
|
x[1].index = 2; |
|
|
|
x[2].index = -1; |
|
|
|
|
|
|
|
for (i = 0; i < XLEN; i++) |
|
|
|
for (j = 0; j < YLEN ; j++) { |
|
|
|
x[0].value = (double) i / XLEN; |
|
|
|
x[1].value = (double) j / YLEN; |
|
|
|
double d = svm_predict(model, x); |
|
|
|
if (param.svm_type == ONE_CLASS && d<0) d=2; |
|
|
|
buffer_painter.setPen(colors[(int)d]); |
|
|
|
buffer_painter.drawPoint(i,j); |
|
|
|
} |
|
|
|
|
|
|
|
svm_free_and_destroy_model(&model); |
|
|
|
delete[] x_space; |
|
|
|
delete[] prob.x; |
|
|
|
delete[] prob.y; |
|
|
|
} |
|
|
|
free(param.weight_label); |
|
|
|
free(param.weight); |
|
|
|
draw_all_points(); |
|
|
|
} |
|
|
|
void button_clear_clicked() |
|
|
|
{ |
|
|
|
clear_all(); |
|
|
|
} |
|
|
|
void button_save_clicked() |
|
|
|
{ |
|
|
|
QString filename = QFileDialog::getSaveFileName(); |
|
|
|
if(!filename.isNull()) |
|
|
|
{ |
|
|
|
FILE *fp = fopen(filename.toAscii().constData(),"w"); |
|
|
|
|
|
|
|
const char *p = input_line.text().toAscii().constData(); |
|
|
|
const char* svm_type_str = strstr(p, "-s "); |
|
|
|
int svm_type = C_SVC; |
|
|
|
if(svm_type_str != NULL) |
|
|
|
sscanf(svm_type_str, "-s %d", &svm_type); |
|
|
|
|
|
|
|
if(fp) |
|
|
|
{ |
|
|
|
if(svm_type == EPSILON_SVR || svm_type == NU_SVR) |
|
|
|
{ |
|
|
|
for(list<point>::iterator p = point_list.begin(); p != point_list.end();p++) |
|
|
|
fprintf(fp,"%f 1:%f\n", p->y, p->x); |
|
|
|
} |
|
|
|
else |
|
|
|
{ |
|
|
|
for(list<point>::iterator p = point_list.begin(); p != point_list.end();p++) |
|
|
|
fprintf(fp,"%d 1:%f 2:%f\n", p->value, p->x, p->y); |
|
|
|
} |
|
|
|
fclose(fp); |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
void button_load_clicked() |
|
|
|
{ |
|
|
|
QString filename = QFileDialog::getOpenFileName(); |
|
|
|
if(!filename.isNull()) |
|
|
|
{ |
|
|
|
FILE *fp = fopen(filename.toAscii().constData(),"r"); |
|
|
|
if(fp) |
|
|
|
{ |
|
|
|
clear_all(); |
|
|
|
char buf[4096]; |
|
|
|
while(fgets(buf,sizeof(buf),fp)) |
|
|
|
{ |
|
|
|
int v; |
|
|
|
double x,y; |
|
|
|
if(sscanf(buf,"%d%*d:%lf%*d:%lf",&v,&x,&y)==3) |
|
|
|
{ |
|
|
|
point p = {x,y,v}; |
|
|
|
point_list.push_back(p); |
|
|
|
} |
|
|
|
else if(sscanf(buf,"%lf%*d:%lf",&y,&x)==2) |
|
|
|
{ |
|
|
|
point p = {x,y,current_value}; |
|
|
|
point_list.push_back(p); |
|
|
|
} |
|
|
|
else |
|
|
|
break; |
|
|
|
} |
|
|
|
fclose(fp); |
|
|
|
draw_all_points(); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
} |
|
|
|
}; |
|
|
|
|
|
|
|
#include "svm-toy.moc" |
|
|
|
|
|
|
|
SvmToyWindow::SvmToyWindow() |
|
|
|
:button_change_icon(this) |
|
|
|
,button_run("Run",this) |
|
|
|
,button_clear("Clear",this) |
|
|
|
,button_save("Save",this) |
|
|
|
,button_load("Load",this) |
|
|
|
,input_line(this) |
|
|
|
,current_value(1) |
|
|
|
{ |
|
|
|
buffer = QPixmap(XLEN,YLEN); |
|
|
|
buffer.fill(Qt::black); |
|
|
|
|
|
|
|
buffer_painter.begin(&buffer); |
|
|
|
|
|
|
|
QObject::connect(&button_change_icon, SIGNAL(clicked()), this, |
|
|
|
SLOT(button_change_icon_clicked())); |
|
|
|
QObject::connect(&button_run, SIGNAL(clicked()), this, |
|
|
|
SLOT(button_run_clicked())); |
|
|
|
QObject::connect(&button_clear, SIGNAL(clicked()), this, |
|
|
|
SLOT(button_clear_clicked())); |
|
|
|
QObject::connect(&button_save, SIGNAL(clicked()), this, |
|
|
|
SLOT(button_save_clicked())); |
|
|
|
QObject::connect(&button_load, SIGNAL(clicked()), this, |
|
|
|
SLOT(button_load_clicked())); |
|
|
|
QObject::connect(&input_line, SIGNAL(returnPressed()), this, |
|
|
|
SLOT(button_run_clicked())); |
|
|
|
|
|
|
|
// don't blank the window before repainting |
|
|
|
setAttribute(Qt::WA_NoBackground); |
|
|
|
|
|
|
|
icon1 = QPixmap(4,4); |
|
|
|
icon2 = QPixmap(4,4); |
|
|
|
icon3 = QPixmap(4,4); |
|
|
|
|
|
|
|
|
|
|
|
QPainter painter; |
|
|
|
painter.begin(&icon1); |
|
|
|
painter.fillRect(0,0,4,4,QBrush(colors[4])); |
|
|
|
painter.end(); |
|
|
|
|
|
|
|
painter.begin(&icon2); |
|
|
|
painter.fillRect(0,0,4,4,QBrush(colors[5])); |
|
|
|
painter.end(); |
|
|
|
|
|
|
|
painter.begin(&icon3); |
|
|
|
painter.fillRect(0,0,4,4,QBrush(colors[6])); |
|
|
|
painter.end(); |
|
|
|
|
|
|
|
button_change_icon.setGeometry( 0, YLEN, 50, 25 ); |
|
|
|
button_run.setGeometry( 50, YLEN, 50, 25 ); |
|
|
|
button_clear.setGeometry( 100, YLEN, 50, 25 ); |
|
|
|
button_save.setGeometry( 150, YLEN, 50, 25); |
|
|
|
button_load.setGeometry( 200, YLEN, 50, 25); |
|
|
|
input_line.setGeometry( 250, YLEN, 250, 25); |
|
|
|
|
|
|
|
input_line.setText(DEFAULT_PARAM); |
|
|
|
button_change_icon.setIcon(icon1); |
|
|
|
} |
|
|
|
|
|
|
|
SvmToyWindow::~SvmToyWindow() |
|
|
|
{ |
|
|
|
buffer_painter.end(); |
|
|
|
} |
|
|
|
|
|
|
|
void SvmToyWindow::mousePressEvent( QMouseEvent* event ) |
|
|
|
{ |
|
|
|
point p = {(double)event->x()/XLEN, (double)event->y()/YLEN, current_value}; |
|
|
|
point_list.push_back(p); |
|
|
|
draw_point(p); |
|
|
|
} |
|
|
|
|
|
|
|
void SvmToyWindow::paintEvent( QPaintEvent* ) |
|
|
|
{ |
|
|
|
// copy the image from the buffer pixmap to the window |
|
|
|
QPainter p(this); |
|
|
|
p.drawPixmap(0, 0, buffer); |
|
|
|
} |
|
|
|
|
|
|
|
int main( int argc, char* argv[] ) |
|
|
|
{ |
|
|
|
QApplication myapp( argc, argv ); |
|
|
|
|
|
|
|
SvmToyWindow* mywidget = new SvmToyWindow(); |
|
|
|
mywidget->setGeometry( 100, 100, XLEN, YLEN+25 ); |
|
|
|
|
|
|
|
mywidget->show(); |
|
|
|
return myapp.exec(); |
|
|
|
} |