non_missing_fix_X = np.logical_not(mask_fit_X)