Skip to content

Ridge Regression Weights

Latest
Compare
Choose a tag to compare
@chrisyeh96 chrisyeh96 released this 02 Jul 08:04

This release includes the weights for the ridge regression models trained on top of the CNNs.

For the DHS out-of-country models...

  • The keys are: ['angola_w', 'angola_b', 'benin_w', 'benin_b', 'burkina_faso_w', 'burkina_faso_b', 'cameroon_w', 'cameroon_b', 'cote_d_ivoire_w', 'cote_d_ivoire_b', 'democratic_republic_of_congo_w', 'democratic_republic_of_congo_b', 'ethiopia_w', 'ethiopia_b', 'ghana_w', 'ghana_b', 'guinea_w', 'guinea_b', 'kenya_w', 'kenya_b', 'lesotho_w', 'lesotho_b', 'malawi_w', 'malawi_b', 'mali_w', 'mali_b', 'mozambique_w', 'mozambique_b', 'nigeria_w', 'nigeria_b', 'rwanda_w', 'rwanda_b', 'senegal_w', 'senegal_b', 'sierra_leone_w', 'sierra_leone_b', 'tanzania_w', 'tanzania_b', 'togo_w', 'togo_b', 'uganda_w', 'uganda_b', 'zambia_w', 'zambia_b', 'zimbabwe_w', 'zimbabwe_b']
  • The out-of-country ridge weights were trained via "leave-one-country-out" cross-validation. Thus, the 'angola_w'/'angola_b' weights should be used for the points from Angola, and similarly for the other countries.

For the DHS incountry models...

  • The keys are: ['A_w', 'A_b', 'B_w', 'B_b', 'C_w', 'C_b', 'D_w', 'D_b', 'E_w', 'E_b']
  • The incountry ridge weights were trained via "leave-one-fold-out" cross-validation. Thus, the 'A_w'/'A_b' weights should be used for the points that belonged in "A" fold, and similarly for the other folds.

Weight dimensions:

  • For the ms and nl models, the ridge regression model has dimension d=512, plus 1 bias term.
  • For the msnl_concat models, the ridge regression model has dimension d=1024, plus 1 bias term.
    • '*_w' is a d-dimensional np.ndarray corresponding to the d-dimensional linear coefficients, and '*_b' is a (1,)-shape np.ndarray corresponding to the bias term

LSMS model ridge regression weights will be added here at a later time.