Commit ee55e8d6 authored by Vlad Shakhuro's avatar Vlad Shakhuro

Hide debug output in Stump

parent a4958fbf
...@@ -22,10 +22,6 @@ static void cumsum(const Mat_<float>& src, Mat_<float> dst) ...@@ -22,10 +22,6 @@ static void cumsum(const Mat_<float>& src, Mat_<float> dst)
} }
} }
#include <iostream>
using std::cout;
using std::endl;
int Stump::train(const Mat& data, const Mat& labels, const Mat& weights) int Stump::train(const Mat& data, const Mat& labels, const Mat& weights)
{ {
CV_Assert(labels.rows == 1 && labels.cols == data.cols); CV_Assert(labels.rows == 1 && labels.cols == data.cols);
...@@ -94,15 +90,6 @@ int Stump::train(const Mat& data, const Mat& labels, const Mat& weights) ...@@ -94,15 +90,6 @@ int Stump::train(const Mat& data, const Mat& labels, const Mat& weights)
neg_total_weight += weights.at<float>(0, col); neg_total_weight += weights.at<float>(0, col);
} }
cout << pos_total_weight << endl;
cout << neg_total_weight << endl;
cout << pos_weights << endl;
cout << neg_weights << endl;
cout << pos_cum_weights << endl;
cout << neg_cum_weights << endl;
/* Compute minimal error */ /* Compute minimal error */
float min_err = FLT_MAX; float min_err = FLT_MAX;
int min_row = -1; int min_row = -1;
...@@ -117,8 +104,6 @@ int Stump::train(const Mat& data, const Mat& labels, const Mat& weights) ...@@ -117,8 +104,6 @@ int Stump::train(const Mat& data, const Mat& labels, const Mat& weights)
err = pos_cum_weights.at<float>(row, col) + err = pos_cum_weights.at<float>(row, col) +
(neg_total_weight - neg_cum_weights.at<float>(row, col)); (neg_total_weight - neg_cum_weights.at<float>(row, col));
cout << "row " << row << "err " << err << endl;
if( err < min_err ) if( err < min_err )
{ {
min_err = err; min_err = err;
...@@ -131,8 +116,6 @@ int Stump::train(const Mat& data, const Mat& labels, const Mat& weights) ...@@ -131,8 +116,6 @@ int Stump::train(const Mat& data, const Mat& labels, const Mat& weights)
err = (pos_total_weight - pos_cum_weights.at<float>(row, col)) + err = (pos_total_weight - pos_cum_weights.at<float>(row, col)) +
neg_cum_weights.at<float>(row, col); neg_cum_weights.at<float>(row, col);
cout << "row " << row << "err " << err << endl;
if( err < min_err ) if( err < min_err )
{ {
min_err = err; min_err = err;
...@@ -143,11 +126,9 @@ int Stump::train(const Mat& data, const Mat& labels, const Mat& weights) ...@@ -143,11 +126,9 @@ int Stump::train(const Mat& data, const Mat& labels, const Mat& weights)
} }
} }
cout << "min_err: " << min_err << endl;
/* Compute threshold, store found values in fields */ /* Compute threshold, store found values in fields */
threshold_ = ( sorted_data.at<int>(min_row, min_col) + threshold_ = ( sorted_data.at<int>(min_row, min_col) +
sorted_data.at<int>(min_row, min_col + 1) ) / 2; sorted_data.at<int>(min_row, min_col + 1) ) / 2;
cout << "threshold: " << threshold_ << endl;
polarity_ = min_polarity; polarity_ = min_polarity;
return min_row; return min_row;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment