database/mariadb-columnstore-engine
1
0
Fork 0
mirror of https://github.com/mariadb-corporation/mariadb-columnstore-engine.git synced 2025-07-30 19:23:07 +03:00
Code Activity

MCOL-521 add regr_intecept and regr_r2

Browse Source
This commit is contained in:
David Hall
2018-09-28 13:51:43 -05:00
parent b8bf311c51
commit dc9ba90f96
11 changed files with 1058 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
utils/regr/regr_slope.cpp
View File

@ -60,11 +60,8 @@ mcsv1_UDAF::ReturnCode regr_slope::init(mcsv1Context* context,
context->setUserDataSize(sizeof(regr_slope_data));
context->setResultType(CalpontSystemCatalog::DOUBLE);
context->setColWidth(8);
if (colTypes[0].scale)
{
context->setScale(colTypes[0].scale + 8);
context->setPrecision(19);
}
context->setScale(colTypes[0].scale + 8);
context->setPrecision(19);
context->setRunFlag(mcsv1sdk::UDAF_IGNORE_NULLS);
return mcsv1_UDAF::SUCCESS;
@ -77,6 +74,7 @@ mcsv1_UDAF::ReturnCode regr_slope::reset(mcsv1Context* context)
data->sumx = 0.0;
data->sumx2 = 0.0;
data->sumy = 0.0;
data->sumxy = 0.0;
return mcsv1_UDAF::SUCCESS;
}
@ -150,7 +148,8 @@ mcsv1_UDAF::ReturnCode regr_slope::evaluate(mcsv1Context* context, static_any::a
double variance = (N * sumx2) - (sumx * sumx);
if (variance != 0)
{
valOut = ((N * sumxy) - (sumx * sumy)) / variance;
double slope = ((N * sumxy) - (sumx * sumy)) / variance;
valOut = slope;
}
}
return mcsv1_UDAF::SUCCESS;