pintos-os.org Git - pspp/blob - src/cat.h

   1 /* PSPP - linear regression.
   2    Copyright (C) 2005 Free Software Foundation, Inc.
   3    Written by Jason H Stover <jason@sakla.net>.
   4
   5    This program is free software; you can redistribute it and/or
   6    modify it under the terms of the GNU General Public License as
   7    published by the Free Software Foundation; either version 2 of the
   8    License, or (at your option) any later version.
   9
  10    This program is distributed in the hope that it will be useful, but
  11    WITHOUT ANY WARRANTY; without even the implied warranty of
  12    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13    General Public License for more details.
  14
  15    You should have received a copy of the GNU General Public License
  16    along with this program; if not, write to the Free Software
  17    Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  18    02110-1301, USA. */
  19
  20 /*
  21   Functions and data structures to recode categorical variables into
  22   vectors and sub-rows of matrices.
  23
  24   To fit many types of statistical models, it is necessary
  25   to change each value of a categorical variable to a vector with binary
  26   entries. These vectors are then stored as sub-rows within a matrix
  27   during model-fitting. We need functions and data strucutres to,
  28   e.g., map a value, say 'a', of a variable named 'cat_var', to a
  29   vector, say (0 1 0 0 0), and vice versa.  We also need to be able
  30   to map the vector back to the value 'a', and if the vector is a
  31   sub-row of a matrix, we need to know which sub-row corresponds to
  32   the variable 'cat_var'.
  33
  34   The data structures defined here will be placed in the variable
  35   structure in the future. When that happens, the useful code
  36   in this file will be that which refers to design matrices.
  37  */
  38
  39 #ifndef CAT_H
  40 #define CAT_H 1
  41
  42 #include <gsl/gsl_matrix.h>
  43 #include <stdbool.h>
  44 /*
  45   This structure contains the observed values of a
  46   categorical variable.
  47  */
  48 struct cat_vals
  49 {
  50   union value *vals;
  51   size_t n_categories;
  52   size_t n_allocated_categories;        /* This is used only during
  53                                            initialization to keep
  54                                            track of the number of
  55                                            values stored.
  56                                          */
  57 };
  58
  59 /*
  60   There are usually multiple categorical variables to recode.  Get rid
  61   of this structure immediately when the variable structure has been
  62   modified to contain the binary encoding.
  63  */
  64 struct recoded_categorical_array
  65 {
  66   struct recoded_categorical **a;
  67   size_t n_vars;
  68 };
  69 /*
  70   The design matrix structure holds the design
  71   matrix and an array to tell us which columns
  72   correspond to which variables. This structure
  73   is not restricted to categorical variables, and
  74   perhaps should be moved to its own module.
  75 */
  76
  77 struct design_matrix_var
  78 {
  79   int first_column;             /* First column for this variable in
  80                                    the design_matix. If this variable
  81                                    is categorical, its values are
  82                                    stored in multiple, contiguous
  83                                    columns, as dictated by its vector
  84                                    encoding in the variable's struct
  85                                    recoded_categorical.
  86                                  */
  87   int last_column;
  88   const struct variable *v;
  89 };
  90 struct design_matrix
  91 {
  92   gsl_matrix *m;
  93   struct design_matrix_var *vars;       /* Element i corresponds to
  94                                            the variable whose values
  95                                            are stored in at least one
  96                                            column of m. If that
  97                                            variable is categorical
  98                                            with more than two
  99                                            categories, its values are
 100                                            stored in multiple,
 101                                            contiguous columns. The
 102                                            variable's values are then
 103                                            stored in the columns
 104                                            first_column through
 105                                            last_column of the
 106                                            design_matrix_var
 107                                            structure.
 108                                          */
 109   size_t n_vars;
 110 };
 111 union value *cat_vector_to_value (const gsl_vector *, struct variable *);
 112
 113 void cat_stored_values_create (struct variable *);
 114
 115 void cat_value_update (struct variable *, const union value *);
 116
 117 int cat_free_recoded_array (struct recoded_categorical_array *);
 118
 119 struct recoded_categorical_array *cr_recoded_cat_ar_create (int,
 120                                                             struct variable
 121                                                             *[]);
 122
 123 void cat_recoded_categorical_create (struct variable *);
 124
 125 void cat_create_value_matrix (struct variable *);
 126
 127 struct recoded_categorical *cat_var_to_recoded_categorical (const struct
 128                                                             variable *,
 129                                                             struct
 130                                                             recoded_categorical_array
 131                                                             *);
 132
 133 struct design_matrix *design_matrix_create (int, const struct variable *[],
 134                                             const size_t);
 135
 136 void design_matrix_destroy (struct design_matrix *);
 137
 138 void design_matrix_set_categorical (struct design_matrix *, size_t,
 139                                     const struct variable *,
 140                                     const union value *);
 141
 142 void design_matrix_set_numeric (struct design_matrix *, size_t,
 143                                 const struct variable *, const union value *);
 144
 145 size_t design_matrix_var_to_column (const struct design_matrix *,
 146                                     const struct variable *);
 147
 148 struct variable *design_matrix_col_to_var (const struct design_matrix *,
 149                                            size_t);
 150
 151 void
 152 design_matrix_set (struct design_matrix *, size_t,
 153                    const struct variable *, const union value *,
 154                    struct recoded_categorical *);
 155
 156 void cat_stored_values_destroy (struct variable *);
 157
 158 #endif