yatsm.regression.design module

Functions for working with statistical model design matrices

Function design_to_indices is used to convert between coefficient types listed in design_coefs and the indices of a design matrix containing the desired coefficient type or types.

yatsm.regression.design.design_to_indices(design_matrix, features)[source]

Return indices of coefficients for features in design matrix

Parameters:
  • design_matrix – OrderedDict containing design features keys and indices of coefficient matrix as values
  • features – list of feature coefficients to extract
Returns:

list of indices and names for each feature specified in

features

Return type:

tuple