uf3.regression.least_squares.get_freezing_mask

get_freezing_mask(n_feats: int, col_idx: numpy.ndarray) → numpy.ndarray

Freezing mask is the set difference between the range of feature indices and the indices to be excluded (col_idx).

Parameters

• n_feats (int) – number of features.

• col_idx (list) – list of indices to be masked.

Returns

set of non-frozen indices.

Return type

mask (np.ndarray)