I am using a very simple keras model in TFX, to solve a regression problem. It seems that TFX wants that you use a keras model with named outputs, so I made: