pivot-table: Kluge dropping default 9-point font to make output readable.
[pspp] / src / language / stats / regression.c
index 3e0871df9eb176f819ccfad6ae2917f33aaa2f6b..28618f12b0a89a2a04a04b10d2f8bd6425d20cc2 100644 (file)
@@ -1,6 +1,6 @@
 /* PSPP - a program for statistical analysis.
    Copyright (C) 2005, 2009, 2010, 2011, 2012, 2013, 2014,
-   2016, 2017 Free Software Foundation, Inc.
+   2016, 2017, 2019 Free Software Foundation, Inc.
 
    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
@@ -60,7 +60,7 @@
 #define STATS_OUTS   8
 #define STATS_CI    16
 #define STATS_BCOV  32
-#define STATS_COLLIN 64
+#define STATS_TOL   64
 
 #define STATS_DEFAULT  (STATS_R | STATS_COEFF | STATS_ANOVA | STATS_OUTS)
 
@@ -309,9 +309,9 @@ cmd_regression (struct lexer *lexer, struct dataset *ds)
                 {
                  statistics |= STATS_BCOV;
                 }
-              else if (lex_match_id (lexer, "COLLIN"))
+              else if (lex_match_id (lexer, "TOL"))
                 {
-                 statistics |= STATS_COLLIN;
+                 statistics |= STATS_TOL;
                 }
               else if (lex_match_id (lexer, "CI"))
                 {
@@ -649,32 +649,36 @@ fill_covariance (gsl_matrix * cov, struct covariance *all_cov,
 
 \f
 
+struct model_container
+{
+  struct linreg **models;
+};
+
 /*
   STATISTICS subcommand output functions.
 */
 static void reg_stats_r (const struct linreg *,     const struct variable *);
-static void reg_stats_coeff (const struct regression *, const struct regression_workspace *,
-                            const struct linreg *, const struct linreg *,
-                            const gsl_matrix *, const struct variable *);
+static void reg_stats_coeff (const struct regression *, const struct linreg *,
+                            const struct model_container *, const gsl_matrix *,
+                            const struct variable *);
 static void reg_stats_anova (const struct linreg *, const struct variable *);
 static void reg_stats_bcov (const struct linreg *,  const struct variable *);
 
 
 static struct linreg **
 run_regression_get_models (const struct regression *cmd,
-                          struct regression_workspace *ws,
                           struct casereader *input,
                           bool output)
 {
   size_t i;
   struct linreg **models = NULL;
-  struct linreg **models_x = NULL;
+  struct model_container *model_container = xzalloc (sizeof (*model_container) * cmd->n_vars);
 
   struct ccase *c;
   struct covariance *cov;
   struct casereader *reader;
 
-  if (cmd->stats & STATS_COLLIN)
+  if (cmd->stats & STATS_TOL)
     {
       for (i = 0; i < cmd->n_vars; i++)
        {
@@ -692,16 +696,10 @@ run_regression_get_models (const struct regression *cmd,
          subreg.resid = false;
          subreg.pred = false;
 
-         struct regression_workspace subws;
-         subws.extras = 0;
-         subws.res_idx = -1;
-         subws.pred_idx = -1;
-         subws.writer = NULL;
-         subws.reader = NULL;
-         subws.residvars = NULL;
-         subws.predvars = NULL;
-
-         models_x = run_regression_get_models (&subreg, &subws, input, false);
+         model_container[i].models =
+           run_regression_get_models (&subreg, input, false);
+         free (subreg.vars);
+         free (subreg.dep_vars);
        }
     }
 
@@ -718,9 +716,7 @@ run_regression_get_models (const struct regression *cmd,
   reader = casereader_clone (input);
   reader = casereader_create_filter_missing (reader, all_vars, n_all_vars,
                                              MV_ANY, NULL, NULL);
-
-
-  {
+{
     struct casereader *r = casereader_clone (reader);
 
     for (; (c = casereader_read (r)) != NULL; case_unref (c))
@@ -762,32 +758,33 @@ run_regression_get_models (const struct regression *cmd,
                reg_stats_anova (models[k], dep_var);
 
              if (cmd->stats & STATS_COEFF)
-               reg_stats_coeff (cmd, ws, models[k],
-                                models_x ? models_x[k] : NULL,
+               reg_stats_coeff (cmd, models[k],
+                                model_container,
                                 cov_matrix, dep_var);
 
              if (cmd->stats & STATS_BCOV)
                reg_stats_bcov  (models[k], dep_var);
-            }
+           }
         }
       else
         {
           msg (SE, _("No valid data found. This command was skipped."));
         }
       free (vars);
+      gsl_matrix_free (cov_matrix);
     }
 
-
   casereader_destroy (reader);
 
-
-  if (models_x)
+  for (int i = 0; i < cmd->n_vars; i++)
     {
-      for (int k = 0; k < cmd->n_dep_vars; k++)
-       linreg_unref (models_x[k]);
-
-      free (models_x);
+      if (model_container[i].models)
+       {
+         linreg_unref (model_container[i].models[0]);
+       }
+      free (model_container[i].models);
     }
+  free (model_container);
 
   free (all_vars);
   free (means);
@@ -800,7 +797,7 @@ run_regression (const struct regression *cmd,
                 struct regression_workspace *ws,
                 struct casereader *input)
 {
-  struct linreg **models = run_regression_get_models (cmd, ws, input, true);
+  struct linreg **models = run_regression_get_models (cmd, input, true);
 
   if (ws->extras > 0)
    {
@@ -859,7 +856,8 @@ reg_stats_r (const struct linreg * c, const struct variable *var)
 {
   struct pivot_table *table = pivot_table_create__ (
     pivot_value_new_text_format (N_("Model Summary (%s)"),
-                                 var_to_string (var)));
+                                 var_to_string (var)),
+    "Model Summary");
 
   pivot_dimension_create (table, PIVOT_AXIS_COLUMN, N_("Statistics"),
                           N_("R"), N_("R Square"), N_("Adjusted R Square"),
@@ -884,13 +882,13 @@ reg_stats_r (const struct linreg * c, const struct variable *var)
   Table showing estimated regression coefficients.
 */
 static void
-reg_stats_coeff (const struct regression *cmd, const struct regression_workspace *ws,
-                const struct linreg *c, const struct linreg *c_x,
-                const gsl_matrix *cov, const struct variable *var)
+reg_stats_coeff (const struct regression *cmd, const struct linreg *c,
+                const struct model_container *mc, const gsl_matrix *cov,
+                const struct variable *var)
 {
   struct pivot_table *table = pivot_table_create__ (
-    pivot_value_new_text_format (N_("Coefficients (%s)"),
-                                 var_to_string (var)));
+    pivot_value_new_text_format (N_("Coefficients (%s)"), var_to_string (var)),
+    "Coefficients");
 
   struct pivot_dimension *statistics = pivot_dimension_create (
     table, PIVOT_AXIS_COLUMN, N_("Statistics"));
@@ -911,7 +909,7 @@ reg_stats_coeff (const struct regression *cmd, const struct regression_workspace
                                     N_("Upper Bound"));
     }
 
-  if (cmd->stats & STATS_COLLIN)
+  if (cmd->stats & STATS_TOL)
     pivot_category_create_group (statistics->root,
                                 N_("Collinearity Statistics"),
                                 N_("Tolerance"), N_("VIF"));
@@ -993,19 +991,14 @@ reg_stats_coeff (const struct regression *cmd, const struct regression_workspace
                              pivot_value_new_number (interval_entries[i]));
        }
 
-      if (cmd->stats & STATS_COLLIN)
+      if (cmd->stats & STATS_TOL)
        {
-         assert (c_x);
-         double rsq = linreg_ssreg (c_x) / linreg_sst (c_x);
-
-         double collin_entries[] = {
-           1.0 - rsq,
-           1.0 / (1.0 - rsq),
-         };
-
-         for (size_t i = 0; i < sizeof collin_entries / sizeof *collin_entries; i++)
-           pivot_table_put2 (table, col++, var_idx,
-                             pivot_value_new_number (collin_entries[i]));
+         {
+           struct linreg *m = mc[j].models[0];
+           double rsq = linreg_ssreg (m) / linreg_sst (m);
+           pivot_table_put2 (table, col++, var_idx, pivot_value_new_number (1.0 - rsq));
+           pivot_table_put2 (table, col++, var_idx, pivot_value_new_number (1.0 / (1.0 - rsq)));
+         }
        }
     }
 
@@ -1019,7 +1012,8 @@ static void
 reg_stats_anova (const struct linreg * c, const struct variable *var)
 {
   struct pivot_table *table = pivot_table_create__ (
-    pivot_value_new_text_format (N_("ANOVA (%s)"), var_to_string (var)));
+    pivot_value_new_text_format (N_("ANOVA (%s)"), var_to_string (var)),
+    "ANOVA");
 
   pivot_dimension_create (table, PIVOT_AXIS_COLUMN, N_("Statistics"),
                           N_("Sum of Squares"), PIVOT_RC_OTHER,
@@ -1074,7 +1068,8 @@ reg_stats_bcov (const struct linreg * c, const struct variable *var)
 {
   struct pivot_table *table = pivot_table_create__ (
     pivot_value_new_text_format (N_("Coefficient Correlations (%s)"),
-                                 var_to_string (var)));
+                                 var_to_string (var)),
+    "Coefficient Correlations");
 
   for (size_t i = 0; i < 2; i++)
     {