X-Git-Url: https://pintos-os.org/cgi-bin/gitweb.cgi?a=blobdiff_plain;f=src%2Fmath%2Flinreg%2Fpredict.c;h=7f404fcf5e575ff212ae7520fc7370af71e17e41;hb=13d1b533c6ce0545a4b06e5e30c73211591034c5;hp=0bab23c8cd89f0b9bcd66ac6fb1cf69cbce72a7b;hpb=fcd8d5f89fe81347a0e4ab4e5256fb89d6bb61c1;p=pspp-builds.git diff --git a/src/math/linreg/predict.c b/src/math/linreg/predict.c index 0bab23c8..7f404fcf 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,13 +31,13 @@ */ double pspp_linreg_predict (const struct variable **predictors, - const union value **vals, - const pspp_linreg_cache * 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; @@ -50,27 +51,27 @@ 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++) { coe = pspp_linreg_get_coeff (c, predictors[j], vals[j]); i = 1; - while (found[i] != coe && i < c->n_coeffs) + while (found[i] == coe && i < c->n_coeffs) { i++; } 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; } @@ -85,8 +86,7 @@ pspp_linreg_predict (const struct variable **predictors, double pspp_linreg_residual (const struct variable **predictors, const union value **vals, - const union value *obs, - const pspp_linreg_cache * c, int n_vals) + const union value *obs, const void *c, int n_vals) { double pred; double result;