X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=src%2Fmath%2Flinreg%2Fpredict.c;h=7f404fcf5e575ff212ae7520fc7370af71e17e41;hb=470affdb48180aebfa764941bbdb353062978444;hp=34e709c82782e039a4c2d6f19b218fda21c64f52;hpb=e947f08aea815968ef993d572977e271fa3ad4da;p=pspp diff --git a/src/math/linreg/predict.c b/src/math/linreg/predict.c index 34e709c827..7f404fcf5e 100644 --- a/src/math/linreg/predict.c +++ b/src/math/linreg/predict.c @@ -18,8 +18,9 @@ Franklin Street, Fifth Floor, Boston, MA 02111-1307, USA. */ +#include #include -#include +#include #include /* @@ -30,14 +31,13 @@ */ double pspp_linreg_predict (const struct variable **predictors, - const union value **vals, - const void *c_, int n_vals) + const union value **vals, const void *c_, int n_vals) { const pspp_linreg_cache *c = c_; int i; int j; - const struct pspp_linreg_coeff **found; - const struct pspp_linreg_coeff *coe; + const struct pspp_coeff **found; + const struct pspp_coeff *coe; double result; double tmp; @@ -51,13 +51,13 @@ pspp_linreg_predict (const struct variable **predictors, return c->depvar_mean; } found = xnmalloc (c->n_coeffs, sizeof (*found)); - *found = c->coeff; - result = c->coeff->estimate; /* Intercept. */ + *found = c->coeff[0]; + result = c->coeff[0]->estimate; /* Intercept. */ /* - The loops guard against the possibility that the caller passed us - inadequate information, such as too few or too many values, or - a redundant list of variable names. + The loops guard against the possibility that the caller passed us + inadequate information, such as too few or too many values, or + a redundant list of variable names. */ for (j = 0; j < n_vals; j++) { @@ -70,8 +70,8 @@ pspp_linreg_predict (const struct variable **predictors, if (i < c->n_coeffs) { found[i] = coe; - tmp = pspp_linreg_coeff_get_est (coe); - if (predictors[j]->type == NUMERIC) + tmp = pspp_coeff_get_est (coe); + if (var_is_numeric (predictors[j])) { tmp *= vals[j]->f; } @@ -86,10 +86,8 @@ pspp_linreg_predict (const struct variable **predictors, double pspp_linreg_residual (const struct variable **predictors, const union value **vals, - const union value *obs, - const void *c_, int n_vals) + const union value *obs, const void *c, int n_vals) { - const pspp_linreg_cache *c = c_; double pred; double result;