diff options
Diffstat (limited to 'src/logistic.h')
| -rw-r--r-- | src/logistic.h | 52 |
1 files changed, 28 insertions, 24 deletions
diff --git a/src/logistic.h b/src/logistic.h index e951935..7f9e133 100644 --- a/src/logistic.h +++ b/src/logistic.h @@ -1,4 +1,4 @@ -#ifndef LOGISTIC_H_ /* Include guard */
+#ifndef LOGISTIC_H_
#define LOGISTIC_H_
// Mixed interface.
@@ -29,19 +29,21 @@ double fLogit_mixed(gsl_vector *beta, double lambdaL1,
double lambdaL2);
-
// Categorical-only interface.
-void logistic_cat_pred(gsl_vector *beta, // Vector of parameters length = 1 + Sum_k(C_k - 1) + Kc
- gsl_matrix_int *X, //Matrix Nobs x K
- gsl_vector_int *nlev, // Vector with number categories
- gsl_vector *yhat); //Vector of prob. predicted by the logistic
+void logistic_cat_pred(gsl_vector *beta, // Vector of parameters
+ // length = 1+Sum_k(C_k-1)+Kc.
+ gsl_matrix_int *X, // Matrix Nobs x K.
+ gsl_vector_int *nlev, // Vector with number categories.
+ gsl_vector *yhat); // Vector of prob. predicted by
+ // the logistic.
-int logistic_cat_fit(gsl_vector *beta, // Vector of parameters length = 1 + Sum_k(C_k - 1) + Kc
- gsl_matrix_int *X, //Matrix Nobs x K
- gsl_vector_int *nlev, // Vector with number categories
- gsl_vector *y, //Vector of prob. to predict
- double lambdaL1, // Regularization L1 0.0 if not used
- double lambdaL2); // Regularization L2 0.0 if not used
+int logistic_cat_fit(gsl_vector *beta, // Vector of parameters
+ // length = 1+Sum_k(C_k-1)+Kc.
+ gsl_matrix_int *X, // Matrix Nobs x K .
+ gsl_vector_int *nlev, // Vector with number categories.
+ gsl_vector *y, // Vector of prob. to predict.
+ double lambdaL1, // Regularization L1, 0 if not used
+ double lambdaL2); // Regularization L2, 0 if not used
double fLogit_cat(gsl_vector *beta,
gsl_matrix_int *X,
@@ -50,22 +52,24 @@ double fLogit_cat(gsl_vector *beta, double lambdaL1,
double lambdaL2);
-/* Continuous only interface */
-void logistic_cont_pred(gsl_vector *beta // Vector of parameters length = 1 + Sum_k(C_k - 1) + Kc
- ,gsl_matrix *Xc // continuous covariates Matrix Nobs x Kc
- ,gsl_vector *yhat //Vector of prob. predicted by the logistic
- );
+// Continuous-only interface.
+void logistic_cont_pred(gsl_vector *beta, // Vector of parameters
+ // length = 1 + Sum_k(C_k-1) + Kc.
+ gsl_matrix *Xc, // Continuous cov's matrix Nobs x Kc.
+ gsl_vector *yhat);// Vector of prob. predicted
+ // by the logistic.
-int logistic_cont_fit(gsl_vector *beta // Vector of parameters length = 1 + Sum_k(C_k - 1) + Kc
- ,gsl_matrix *Xc // continuous covariates Matrix Nobs x Kc
- ,gsl_vector *y //Vector of prob. to predict
- ,double lambdaL1 // Regularization L1 0.0 if not used
- ,double lambdaL2); // Regularization L2 0.0 if not used
+int logistic_cont_fit(gsl_vector *beta, // Vector of parameters
+ // length = 1+Sum_k(C_k-1)+Kc.
+ gsl_matrix *Xc, // Continuous cov's matrix Nobs x Kc.
+ gsl_vector *y, // Vector of prob. to predict.
+ double lambdaL1, // Regularization L1, 0 if not used.
+ double lambdaL2); // Regularization L2, 0 if not used.
double fLogit_cont(gsl_vector *beta,
- gsl_matrix *Xc, // Continuous covariates matrix Nobs x Kc .
+ gsl_matrix *Xc, // Continuous covariates matrix Nobs x Kc.
gsl_vector *y,
double lambdaL1,
double lambdaL2);
-#endif // LOGISTIC_H_
+#endif
|