Full udders are painful. For humans too. If mom starts breastfeeding and then abruptly stops, the boobs will swell up and ache horribly for several days (until lactation stops due to lack of stimulation.)

You don’t need to train the cow. After it’s milked once with the machine, it associates the thing with pain relief (plus a little snack to reinforce.)