I am trying to calculate the margins for some keras models that I have using tensorflow 2. I saw some code that was written by an AI resident at Google but it was written for te