I am trying to use the Tensorflow guide for loading images but when I tweak it for my use-case I run into an error. I would like to build a regression model and get the valu