Generalization Probe