struct tab_table *t;
assert (c != NULL);
- n_rows = c->n_coeffs + 3;
+ n_rows = linreg_n_coeffs (c) + 3;
t = tab_create (n_cols, n_rows, 0);
tab_headers (t, 2, 0, 1, 0);
int n_trns = 0;
int i;
- assert (models != NULL);
-
if (save)
{
/* Count the number of transformations we will need. */
fill_covariance (gsl_matrix *cov, struct covariance *all_cov,
const struct variable **vars,
size_t n_vars, const struct variable *dep_var,
- const struct variable **all_vars, size_t n_all_vars)
+ const struct variable **all_vars, size_t n_all_vars,
+ double *means)
{
size_t i;
size_t j;
- size_t k = 0;
size_t dep_subscript;
size_t *rows;
const gsl_matrix *ssizes;
const gsl_matrix *cm;
+ const gsl_matrix *mean_matrix;
double result = 0.0;
cm = covariance_calculate (all_cov);
for (i = 0; i < n_all_vars; i++)
{
- for (j = k; j < n_vars; j++)
+ for (j = 0; j < n_vars; j++)
{
if (vars[j] == all_vars[i])
{
- if (vars[j] != dep_var)
- {
- rows[j] = i;
- }
- else
- {
- dep_subscript = i;
- }
- k++;
- break;
+ rows[j] = i;
}
}
+ if (all_vars[i] == dep_var)
+ {
+ dep_subscript = i;
+ }
}
+ mean_matrix = covariance_moments (all_cov, MOMENT_MEAN);
for (i = 0; i < cov->size1 - 1; i++)
{
+ means[i] = gsl_matrix_get (mean_matrix, rows[i], 0);
for (j = 0; j < cov->size2 - 1; j++)
{
gsl_matrix_set (cov, i, j, gsl_matrix_get (cm, rows[i], rows[j]));
gsl_matrix_set (cov, j, i, gsl_matrix_get (cm, rows[j], rows[i]));
}
}
+ means[cov->size1 - 1] = gsl_matrix_get (mean_matrix, dep_subscript, 0);
ssizes = covariance_moments (all_cov, MOMENT_NONE);
result = gsl_matrix_get (ssizes, dep_subscript, rows[0]);
for (i = 0; i < cov->size1 - 1; i++)
}
}
free (rows);
+ return result;
}
static bool
int n_indep = 0;
int k;
double n_data;
+ double *means;
struct ccase *c;
struct covariance *cov;
const struct variable **vars;
dict_get_vars (dict, &v_variables, &n_variables, 0);
}
vars = xnmalloc (n_variables, sizeof (*vars));
+ means = xnmalloc (n_variables, sizeof (*vars));
cov = covariance_1pass_create (n_variables, v_variables,
dict_get_weight (dict), MV_ANY);
this_cm = gsl_matrix_alloc (n_indep + 1, n_indep + 1);
n_data = fill_covariance (this_cm, cov, vars, n_indep,
- dep_var, v_variables, n_variables);
+ dep_var, v_variables, n_variables, means);
models[k] = linreg_alloc (dep_var, (const struct variable **) vars,
n_data, n_indep);
models[k]->depvar = dep_var;
-
+ for (i = 0; i < n_indep; i++)
+ {
+ linreg_set_indep_variable_mean (models[k], i, means[i]);
+ }
/*
For large data sets, use QR decomposition.
*/
{
msg (SE,
gettext ("No valid data found. This command was skipped."));
+ linreg_free (models[k]);
+ models[k] = NULL;
}
}
-
+
casereader_destroy (reader);
free (vars);
+ free (means);
casereader_destroy (input);
covariance_destroy (cov);