[Pspp-cvs] pspp/src/language/stats regression.q ChangeLog

[Jason H Stover]

CVSROOT:        /sources/pspp
Module name:    pspp
Branch:         
Changes by:     Jason H Stover <address@hidden> 06/04/28 19:17:15

Modified files:
        src/language/stats: regression.q ChangeLog 

Log message:
        pass only the variables in the model to pspp_linreg_predict()

CVSWeb URLs:
http://cvs.savannah.gnu.org/viewcvs/pspp/pspp/src/language/stats/regression.q.diff?tr1=1.19&tr2=1.20&r1=text&r2=text
http://cvs.savannah.gnu.org/viewcvs/pspp/pspp/src/language/stats/ChangeLog.diff?tr1=1.16&tr2=1.17&r1=text&r2=text

Patches:
Index: pspp/src/language/stats/ChangeLog
diff -u pspp/src/language/stats/ChangeLog:1.16 
pspp/src/language/stats/ChangeLog:1.17
--- pspp/src/language/stats/ChangeLog:1.16      Fri Apr 28 13:25:53 2006
+++ pspp/src/language/stats/ChangeLog   Fri Apr 28 19:17:15 2006
@@ -3,6 +3,9 @@
        * regression.q (regression_trns_resid_proc): Pass only the
        variables used in the model to (*model->residual)().
 
+       * regression.q (regression_trns_pred_proc): Pass only the
+       variables used in the model to (*model->pred)().
+
 2006-04-26  Jason Stover  <address@hidden>
 
        * regression.q: Added support for multiple transformations.
Index: pspp/src/language/stats/regression.q
diff -u pspp/src/language/stats/regression.q:1.19 
pspp/src/language/stats/regression.q:1.20
--- pspp/src/language/stats/regression.q:1.19   Fri Apr 28 13:25:53 2006
+++ pspp/src/language/stats/regression.q        Fri Apr 28 19:17:15 2006
@@ -539,58 +539,34 @@
 regression_trns_pred_proc (void *t_, struct ccase *c, int case_idx UNUSED)
 {
   size_t i;
-  size_t n_vars;
-  size_t n_vals = 0;
-  union value *tmp = NULL;
-  struct reg_trns *t = t_;
+  size_t n_vals;
+  struct reg_trns *trns = t_;
   pspp_linreg_cache *model;
-  union value *output;
+  union value *output = NULL;
   const union value **vals = NULL;
   struct variable **vars = NULL;
-  struct variable **model_vars = NULL;
   
-  assert (t != NULL);
-  model = t->c;
+  assert (trns!= NULL);
+  model = trns->c;
   assert (model != NULL);
   assert (model->depvar != NULL);
   assert (model->pred != NULL);
-  
-  dict_get_vars (default_dict, &vars, &n_vars, 1u << DC_SYSTEM);
-  vals = xnmalloc (n_vars, sizeof (*vals));
-  model_vars = xnmalloc (n_vars, sizeof (*model_vars));
-  assert (vals != NULL);
+
+  vars = xnmalloc (model->n_coeffs, sizeof (*vars));
+  n_vals = (*model->get_vars) (model, vars);
+
+  vals = xnmalloc (n_vals, sizeof (*vals));
   output = case_data_rw (c, model->pred->fv);
   assert (output != NULL);
 
-  for (i = 0; i < n_vars; i++)
+  for (i = 0; i < n_vals; i++)
     {
-      /* Use neither the predicted values nor the dependent variable. */
-      if (vars[i]->index != model->pred->index &&
-         vars[i]->index != model->depvar->index)
-       {
-         if (vars[i]->type == ALPHA && vars[i]->obs_vals != NULL)
-           {
-             tmp = vars[i]->obs_vals->vals;
-           }
-         else 
-           {
-             tmp = NULL;
-           }
-         /* 
-            Make sure the variable we use is in the linear model.
-         */
-         if (pspp_linreg_get_coeff (model, vars[i], tmp) != NULL)
-           {
-             vals[n_vals] = case_data (c, i);
-             model_vars[n_vals] = vars[i];
-             n_vals++;
-           }
-       }
+      vals[i] = case_data (c, vars[i]->fv);
     }
   output->f = (*model->predict) ((const struct variable **) vars, 
                                  vals, model, n_vals);
   free (vals);
-  free (model_vars);
+  free (vars);
   return TRNS_CONTINUE;
 }
 /*