fixed SVM

This commit is contained in:
Maria Dimashova 2010-08-09 15:43:06 +00:00
parent a2950d5d44
commit 82a98521ed
2 changed files with 3 additions and 1 deletions

View File

@ -676,7 +676,7 @@ cvPreprocessCategoricalResponses( const CvMat* responses,
if( r_type != CV_32FC1 && r_type != CV_32SC1 )
CV_ERROR( CV_StsUnsupportedFormat, "Unsupported response type" );
r_step = responses->step ? responses->step / CV_ELEM_SIZE(responses->type) : 1;
r_step = responses->rows == 1 ? 1 : responses->step / CV_ELEM_SIZE(responses->type);
if( sample_idx )
{

View File

@ -1330,6 +1330,8 @@ bool CvSVM::do_train( int svm_type, int sample_count, int var_count, const float
const int sample_size = var_count*sizeof(samples[0][0]);
int i, j, k;
cvClearMemStorage( storage );
if( svm_type == ONE_CLASS || svm_type == EPS_SVR || svm_type == NU_SVR )
{
int sv_count = 0;