Estimate heritability on subset of individuals