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) – OrderedDict containing design features keys and indices of coefficient matrix as values
- features (list) – list of feature coefficients to extract
Returns: - list of indices and names for each feature specified in
features
Return type: