tf.decode_csv参数“标签”的数据类型float64不在允许的值列表中:int32,int64

时间:2018-09-11 22:56:11

标签: tensorflow

试图将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

1 个答案:

答案 0 :(得分:0)

从头开始,我的.csv文件出错。