Tf.decode_csv параметр 'labels' имеет DataType float64 не в списке допустимых значений: int32, int64

пытаюсь передать TensorFlow tf.decode_csv тип данных float64, но получаю ошибку, это недопустимо

    CSV_TYPES = [[0.0], [2.], [0.0], [0.0], [0.0], [0]]
    def _parse_line(line):
        fields = tf.decode_csv(line, record_defaults=CSV_TYPES)

Проблема заключается во втором значении, которое, например, здесь мои значения

9.5,-110,9.5,50,68,1
9.5,+100,9.5,50,32,0
7,-110,7,30,24,1
7,-120,70,76,0

что вызывает следующую ошибку:

TypeError: Value passed to parameter 'labels' has DataType float64 not in list of allowed values: int32, int64

видел много дискуссий по этой проблеме, но не мог понять, как разрешить меткам иметь DataType float64 специально