// -----------------------------------------------------------------------
// Linear regression
// -----------------------------------------------------------------------
AnyType
linregr_transition::run(AnyType& args) {
MutableLinRegrState state = args[0].getAs<MutableByteString>();
if (args[1].isNull() || args[2].isNull()) { return args[0]; }
double y = args[1].getAs<double>();
MappedColumnVector x;
try {
MappedColumnVector xx = args[2].getAs<MappedColumnVector>();
x.rebind(xx.memoryHandle(), xx.size());
} catch (const ArrayWithNullException &e) {
return args[0];
}
state << MutableLinRegrState::tuple_type(x, y);
return state.storage();
}
AnyType
multi_response_glm_multinom_logit_transition::run(AnyType& args) {
MutableMultiResponseGLMState state = args[0].getAs<MutableByteString>();
if (state.terminated || args[1].isNull() || args[2].isNull()) {
return args[0];
}
double y = args[1].getAs<double>();
MappedColumnVector x;
try {
MappedColumnVector xx = args[2].getAs<MappedColumnVector>();
x.rebind(xx.memoryHandle(), xx.size());
} catch (const ArrayWithNullException &e) {
return args[0];
}
if (state.empty()) {
state.num_features = static_cast<uint16_t>(x.size());
state.num_categories = args[4].getAs<uint16_t>();
state.optimizer.num_coef = static_cast<uint16_t>(
state.num_features * (state.num_categories-1));
// MADLIB-667: GPDB limits the single array size to be 1GB, which means
// that the size of a double array cannot be large than 134217727
// because (134217727 * 8) / (1024 * 1024) = 1023. And solve
// state_size = x^2 + 2^x + 6 <= 134217727 will give x <= 11584.
uint32_t state_size = 6 +
state.optimizer.num_coef * state.optimizer.num_coef +
2 * state.optimizer.num_coef;
if(state_size > 134217727){
throw std::runtime_error(
"The product of number of independent variables and number of "
"categories cannot be larger than 11584.");
}
state.resize();
if (!args[3].isNull()) {
MultiResponseGLMState prev_state = args[3].getAs<ByteString>();
state = prev_state;
state.reset();
}
}
state << MutableMultiResponseGLMState::tuple_type(x, y);
return state.storage();
}
