Just pass label_cls=FloatList. The Points classes are for points, so two coordinates, and more specifically ImagePoints (when you need to apply data augmentation to the points the same way as the image). For all other regression problems, FloatList is the label class you need.
1 Like